diff options
Diffstat (limited to 'tensorflow/core/kernels/crop_and_resize_op.cc')
-rw-r--r-- | tensorflow/core/kernels/crop_and_resize_op.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/crop_and_resize_op.cc b/tensorflow/core/kernels/crop_and_resize_op.cc index 6a748d3462..c5470f81eb 100644 --- a/tensorflow/core/kernels/crop_and_resize_op.cc +++ b/tensorflow/core/kernels/crop_and_resize_op.cc @@ -79,13 +79,13 @@ static inline Status ParseAndCheckBoxSizes(const Tensor& boxes, template <typename Device> inline void RunIfBoxIndexIsValid( OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index, - int batch_size, Callback compute, Callback done); + int batch_size, const Callback& compute, const Callback& done); // Specialization of CheckValidBoxIndex for a CPUDevice. template <> inline void RunIfBoxIndexIsValid<CPUDevice>( OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index, - int batch_size, Callback compute, Callback done) { + int batch_size, const Callback& compute, const Callback& done) { const int num_boxes = box_index.dimension(0); for (int b = 0; b < num_boxes; ++b) { OP_REQUIRES_ASYNC( @@ -690,7 +690,7 @@ namespace { template <> inline void RunIfBoxIndexIsValid<GPUDevice>( OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index, - int batch_size, Callback compute, Callback done) { + int batch_size, const Callback& compute, const Callback& done) { const int num_boxes = box_index.dimension(0); if (num_boxes == 0) { compute(); |