diff options
Diffstat (limited to 'tensorflow/core/kernels/transpose_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/transpose_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/transpose_op.cc b/tensorflow/core/kernels/transpose_op.cc index 88786ec774..5ecef9c6f9 100644 --- a/tensorflow/core/kernels/transpose_op.cc +++ b/tensorflow/core/kernels/transpose_op.cc @@ -169,7 +169,7 @@ Status TransposeGpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, .TypeConstraint<T>("T") \ .HostMemory("perm"), \ TransposeGpuOp); -TF_CALL_NUMBER_TYPES(REGISTER); +TF_CALL_POD_TYPES(REGISTER); #undef REGISTER #endif |