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

Fix cherrypick issue
This commit is contained in:
Mihai Maruseac 2021-06-09 14:20:35 -07:00 committed by GitHub
commit a851bdbcc6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -597,6 +597,7 @@ class SparseCrossOp : public OpKernel {
}
int64 num_buckets_;
uint64 hash_key_;
DataType internal_type_;
};
REGISTER_KERNEL_BUILDER(Name("SparseCross")