Merge pull request #50183 from tensorflow/geetachavan1-patch-1

Fix cherrypick issue
tensorflow · Jun 9, 2021 · a851bdb · a851bdb
2 parents ce8bb42 + e10f338
commit a851bdb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/core/kernels/sparse_cross_op.cc b/tensorflow/core/kernels/sparse_cross_op.cc
@@ -597,6 +597,7 @@ class SparseCrossOp : public OpKernel {
   }
   int64 num_buckets_;
   uint64 hash_key_;
+  DataType internal_type_;
 };
 
 REGISTER_KERNEL_BUILDER(Name("SparseCross")