diff options
author | 2017-01-17 13:38:04 -0800 | |
---|---|---|
committer | 2017-01-17 13:53:35 -0800 | |
commit | bf67d0a1c52c303f9018c00cbc030dc35438ed2c (patch) | |
tree | 8d2e8edd14dd474218245b2b1db5331b107ac763 | |
parent | da0116b01584b0f044b3f36a7543f7a77b66ff8b (diff) |
Remove extra unnecessary parameter to functor::ResizeBilinear
This extra parameter got introduced in a previous commit as part
of some performance improvements. But because it is unused, it
should be reverted back.
Change: 144749662
-rw-r--r-- | tensorflow/core/kernels/resize_bilinear_op.cc | 9 | ||||
-rw-r--r-- | tensorflow/core/kernels/resize_bilinear_op.h | 4 | ||||
-rw-r--r-- | tensorflow/core/kernels/resize_bilinear_op_gpu.cu.cc | 3 |
3 files changed, 6 insertions, 10 deletions
diff --git a/tensorflow/core/kernels/resize_bilinear_op.cc b/tensorflow/core/kernels/resize_bilinear_op.cc index 7630fe1ffc..85d28d2c64 100644 --- a/tensorflow/core/kernels/resize_bilinear_op.cc +++ b/tensorflow/core/kernels/resize_bilinear_op.cc @@ -55,9 +55,9 @@ class ResizeBilinearOp : public OpKernel { typename TTypes<float, 4>::Tensor output_data = st.output->tensor<float, 4>(); - functor::ResizeBilinear<Device, T>()( - context, context->eigen_device<Device>(), image_data, st.height_scale, - st.width_scale, output_data); + functor::ResizeBilinear<Device, T>()(context->eigen_device<Device>(), + image_data, st.height_scale, + st.width_scale, output_data); } private: @@ -263,8 +263,7 @@ inline void scale_similar_image(const float* input_image, const int b, namespace functor { template <typename T> struct ResizeBilinear<CPUDevice, T> { - void operator()(OpKernelContext* context, const CPUDevice& d, - typename TTypes<T, 4>::ConstTensor images, + void operator()(const CPUDevice& d, typename TTypes<T, 4>::ConstTensor images, const float height_scale, const float width_scale, typename TTypes<float, 4>::Tensor output) { const int batch_size = images.dimension(0); diff --git a/tensorflow/core/kernels/resize_bilinear_op.h b/tensorflow/core/kernels/resize_bilinear_op.h index 762d92f0c2..d16ed64f14 100644 --- a/tensorflow/core/kernels/resize_bilinear_op.h +++ b/tensorflow/core/kernels/resize_bilinear_op.h @@ -18,7 +18,6 @@ limitations under the License. #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/core/framework/numeric_types.h" -#include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/tensor_types.h" namespace tensorflow { @@ -26,8 +25,7 @@ namespace functor { template <typename Device, typename T> struct ResizeBilinear { - void operator()(OpKernelContext* context, const Device& d, - typename TTypes<T, 4>::ConstTensor images, + void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor images, const float height_scale, const float width_scale, typename TTypes<float, 4>::Tensor resized_images); }; diff --git a/tensorflow/core/kernels/resize_bilinear_op_gpu.cu.cc b/tensorflow/core/kernels/resize_bilinear_op_gpu.cu.cc index c2dac06364..a7da7a0777 100644 --- a/tensorflow/core/kernels/resize_bilinear_op_gpu.cu.cc +++ b/tensorflow/core/kernels/resize_bilinear_op_gpu.cu.cc @@ -149,8 +149,7 @@ namespace functor { // Partial specialization of ResizeBilinear functor for a GPUDevice. template <typename T> struct ResizeBilinear<GPUDevice, T> { - void operator()(OpKernelContext* context, const GPUDevice& d, - typename TTypes<T, 4>::ConstTensor images, + void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor images, const float height_scale, const float width_scale, typename TTypes<float, 4>::Tensor output) { const int batch = images.dimension(0); |