diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2016-05-26 11:05:13 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-05-26 12:18:27 -0700 |
| commit | 8cc567bf9703e14f0e16595eb9f220859a5eff20 (patch) | |
| tree | 13694c01d607a2d2d2bad57ae533894214eee7f5 /tensorflow/core/kernels/transpose_functor_gpu.cu.cc | |
| parent | 9a69f398e9958d975db0c651e9ec95762b2ff8b4 (diff) | |
Merge changes from github.
Change: 123342870
Diffstat (limited to 'tensorflow/core/kernels/transpose_functor_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/transpose_functor_gpu.cu.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/transpose_functor_gpu.cu.cc b/tensorflow/core/kernels/transpose_functor_gpu.cu.cc index 3febba0441..8e6cbc4270 100644 --- a/tensorflow/core/kernels/transpose_functor_gpu.cu.cc +++ b/tensorflow/core/kernels/transpose_functor_gpu.cu.cc @@ -130,6 +130,10 @@ Status DoTranspose<Device>(const Device& d, const Tensor& in, internal::Transpose<Device, uint64>(d, in, perm, out); break; + case DT_COMPLEX128: + internal::Transpose<Device, float4>(d, in, perm, out); + break; + default: return errors::Unimplemented("Unsupported dtype on GPU: ", in.dtype()); } |