diff options
author | RJ Ryan <rjryan@google.com> | 2016-09-14 11:49:35 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-14 13:01:36 -0700 |
commit | f21642013092b53186491064335053a9e02ce010 (patch) | |
tree | 7fd3328247017507192325b40aeca868774e601b /tensorflow/core/kernels/slice_op_cpu_impl.h | |
parent | bad65c0641a08dd7939cbe5aa1c1e62c3bb78634 (diff) |
Add complex64/complex128 GPU support for the following ops:
* ComplexAbs
* Shape
* ShapeN
* Sign
* Slice
* ZerosLike
Also, add a "special" int32 Reverse GPU kernel to avoid unnecessary CPU/GPU transfers.
Change: 133160947
Diffstat (limited to 'tensorflow/core/kernels/slice_op_cpu_impl.h')
-rw-r--r-- | tensorflow/core/kernels/slice_op_cpu_impl.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/slice_op_cpu_impl.h b/tensorflow/core/kernels/slice_op_cpu_impl.h index ce8e298a9d..0b0700ec36 100644 --- a/tensorflow/core/kernels/slice_op_cpu_impl.h +++ b/tensorflow/core/kernels/slice_op_cpu_impl.h @@ -32,7 +32,7 @@ using CpuDevice = Eigen::ThreadPoolDevice; TF_CALL_ALL_TYPES(DEFINE_CPU_KERNELS); DEFINE_CPU_KERNELS(bfloat16); -#undef DEFINE_GPU_KERNELS +#undef DEFINE_CPU_KERNELS } // namespace tensorflow |