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

Merge pull request from tensorflow/fix-cp-2.0

Browse Source 
Fix bad cherrypick
This commit is contained in:
Mihai Maruseac 2020-12-17 14:30:28 -08:00 committed by GitHub
commit 33f107fa6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

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>(),