diff options
Diffstat (limited to 'tensorflow/core/kernels/sample_distorted_bounding_box_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/sample_distorted_bounding_box_op.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/sample_distorted_bounding_box_op.cc b/tensorflow/core/kernels/sample_distorted_bounding_box_op.cc index 44a817a5c7..c0fde8042e 100644 --- a/tensorflow/core/kernels/sample_distorted_bounding_box_op.cc +++ b/tensorflow/core/kernels/sample_distorted_bounding_box_op.cc @@ -387,9 +387,9 @@ class SampleDistortedBoundingBoxV2Op : public OpKernel { OP_REQUIRES_OK( context, context->allocate_output(2, TensorShape({1, 1, 4}), &bboxes)); - typename TTypes<T, 1>::Tensor begin_data = begin->tensor<T, 1>(); - typename TTypes<T, 1>::Tensor size_data = size->tensor<T, 1>(); - typename TTypes<float, 3>::Tensor bboxes_data = bboxes->tensor<float, 3>(); + typename TTypes<T, 1>::Tensor begin_data(begin->tensor<T, 1>()); + typename TTypes<T, 1>::Tensor size_data(size->tensor<T, 1>()); + TTypes<float, 3>::Tensor bboxes_data = bboxes->tensor<float, 3>(); begin_data(0) = T(offset_height); size_data(0) = T(target_height); |