experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix bad cherrypick

Browse Source
This commit is contained in:
Geeta Chavan 2020-12-17 13:52:36 -08:00
parent d3ea1b93cf
commit d04c0e0219
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/data_format_ops.cc
View File

@ -185,7 +185,7 @@ class DataFormatVecPermuteOp : public OpKernel {
errors::InvalidArgument( errors::InvalidArgument(
"Format specifier must contain H and W for 2D case")); "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<Device, T>()(context->eigen_device<Device>(), functor::DataFormatVecPermute<Device, T>()(context->eigen_device<Device>(),
input.flat<T>(), input.flat<T>(),