diff options
Diffstat (limited to 'tensorflow/contrib/image')
| -rw-r--r-- | tensorflow/contrib/image/kernels/bipartite_match_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/image/kernels/bipartite_match_op.cc b/tensorflow/contrib/image/kernels/bipartite_match_op.cc index 7d207c388b..726adb0777 100644 --- a/tensorflow/contrib/image/kernels/bipartite_match_op.cc +++ b/tensorflow/contrib/image/kernels/bipartite_match_op.cc @@ -85,7 +85,7 @@ class BipartiteMatchOp : public OpKernel { context->allocate_output(1, TensorShape({num_input_columns}), &column_to_row_match_indices)); - typename TTypes<float, 2>::ConstTensor distance_mat = + TTypes<float, 2>::ConstTensor distance_mat = input_distance_mat.shaped<float, 2>( {num_input_rows, num_input_columns}); |