diff --git a/tensorflow/core/kernels/data_format_ops.cc b/tensorflow/core/kernels/data_format_ops.cc index d00e51b5ea5..a51bfc3b172 100644 --- a/tensorflow/core/kernels/data_format_ops.cc +++ b/tensorflow/core/kernels/data_format_ops.cc @@ -185,7 +185,7 @@ class DataFormatVecPermuteOp : public OpKernel { errors::InvalidArgument( "Format specifier must contain H and W for 2D case")); } - ComputeDstIndex(src_format_str, dst_format_str, input.dims(), &dst_idx); + ComputeDstIndex(input.dims(), &dst_idx); functor::DataFormatVecPermute()(context->eigen_device(), input.flat(),