diff --git a/tensorflow/core/kernels/sparse_cross_op.cc b/tensorflow/core/kernels/sparse_cross_op.cc index a508bc76ba5..cf02e89b5ea 100644 --- a/tensorflow/core/kernels/sparse_cross_op.cc +++ b/tensorflow/core/kernels/sparse_cross_op.cc @@ -584,6 +584,7 @@ class SparseCrossOp : public OpKernel { } int64 num_buckets_; uint64 hash_key_; + DataType internal_type_; }; REGISTER_KERNEL_BUILDER(Name("SparseCross")