diff options
author | Olivia Nordquist <nolivia@google.com> | 2016-09-15 11:24:57 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-15 12:32:42 -0700 |
commit | 71b1f3ee6629580e6881fb853275b124faae67be (patch) | |
tree | f44ec747c6eeb057d9db047e09fea7a146fe1d3d /tensorflow/core/kernels/tile_ops_gpu.cu.cc | |
parent | e86469cdd579d33d2beafd0803dc67c77424cc5f (diff) |
enable tile for complex inputs in gpu, github issue #2577
Change: 133293158
Diffstat (limited to 'tensorflow/core/kernels/tile_ops_gpu.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/tile_ops_gpu.cu.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/tile_ops_gpu.cu.cc b/tensorflow/core/kernels/tile_ops_gpu.cu.cc index 6a83ad4b4f..518e91f554 100644 --- a/tensorflow/core/kernels/tile_ops_gpu.cu.cc +++ b/tensorflow/core/kernels/tile_ops_gpu.cu.cc @@ -18,6 +18,8 @@ limitations under the License. #define EIGEN_USE_GPU #include <stdio.h> +#include <complex> +#include "tensorflow/core/framework/numeric_types.h" #include "tensorflow/core/kernels/tile_ops_impl.h" namespace tensorflow { @@ -44,6 +46,8 @@ DEFINE_TYPE(Eigen::half) DEFINE_TYPE(int64) DEFINE_TYPE(int32) DEFINE_TYPE(int16) +DEFINE_TYPE(complex64) +DEFINE_TYPE(complex128) // NOTE(keveman): Eigen's int8 and string versions don't compile yet with nvcc. #undef DEFINE_DIM |