diff options
Diffstat (limited to 'tensorflow/core/kernels/crop_and_resize_op.h')
| -rw-r--r-- | tensorflow/core/kernels/crop_and_resize_op.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/crop_and_resize_op.h b/tensorflow/core/kernels/crop_and_resize_op.h index 9278893704..22df1bdd56 100644 --- a/tensorflow/core/kernels/crop_and_resize_op.h +++ b/tensorflow/core/kernels/crop_and_resize_op.h @@ -26,7 +26,7 @@ namespace functor { template <typename Device, typename T> struct CropAndResize { // We assume that the tensor sizes are correct. - void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor image, + bool operator()(const Device& d, typename TTypes<T, 4>::ConstTensor image, typename TTypes<float, 2>::ConstTensor boxes, typename TTypes<int32, 1>::ConstTensor box_ind, float extrapolation_value, @@ -36,7 +36,7 @@ struct CropAndResize { template <typename Device, typename T> struct CropAndResizeBackpropImage { // We assume that the tensor sizes are correct. - void operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads, + bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads, typename TTypes<float, 2>::ConstTensor boxes, typename TTypes<int32, 1>::ConstTensor box_ind, typename TTypes<T, 4>::Tensor grads_image); @@ -45,7 +45,7 @@ struct CropAndResizeBackpropImage { template <typename Device, typename T> struct CropAndResizeBackpropBoxes { // We assume that the tensor sizes are correct. - void operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads, + bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads, typename TTypes<T, 4>::ConstTensor image, typename TTypes<float, 2>::ConstTensor boxes, typename TTypes<int32, 1>::ConstTensor box_ind, |