OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Update sparse_cross_op.cc

Browse Source
This commit is contained in:
geetachavan1 2021-06-07 12:08:33 -07:00 committed by GitHub
parent e5f7d375b6
commit acba599cdb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/sparse_cross_op.cc
View File

@ -311,7 +311,7 @@ class SparseCrossOp : public OpKernel {
context->input_list("dense_inputs", &dense_list_in)); context->input_list("dense_inputs", &dense_list_in));
DataType internal_type = internal_type_; DataType internal_type = internal_type_;
ValidateInput(context, indices_list_in, values_list_in, ValidateInput(context, indices_list_in, values_list_in,
shapes_list_in, dense_list_in, internal_type)); shapes_list_in, dense_list_in, internal_type);
OP_REQUIRES_OK(context, context->status()); OP_REQUIRES_OK(context, context->status());
std::vector<std::unique_ptr<ColumnInterface<InternalType>>> columns = std::vector<std::unique_ptr<ColumnInterface<InternalType>>> columns =