diff options
| author |  Geoffrey Irving <geoffreyi@google.com> | 2016-03-08 15:25:25 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-08 17:20:51 -0800 |
| commit | 6c576f017156cc0bf50e2dd2cd42b2ab8db9bffe (patch) | |
| tree | 9eafbfa32678f9114ed18066fd99a950051eb29f | |
| parent | e02403c442cc74673b7d24b5e8104d60f6448c5d (diff) | |
Add bool transpose on GPU
Fixes #542.
Change: 116704226
| -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 |