diff options
author | 2016-09-14 11:49:35 -0800 | |
---|---|---|
committer | 2016-09-14 13:01:36 -0700 | |
commit | f21642013092b53186491064335053a9e02ce010 (patch) | |
tree | 7fd3328247017507192325b40aeca868774e601b /tensorflow/core/kernels/cwise_op_abs.cc | |
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/cwise_op_abs.cc')
-rw-r--r-- | tensorflow/core/kernels/cwise_op_abs.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_op_abs.cc b/tensorflow/core/kernels/cwise_op_abs.cc index daa41ecbdf..fbc23b3b6f 100644 --- a/tensorflow/core/kernels/cwise_op_abs.cc +++ b/tensorflow/core/kernels/cwise_op_abs.cc @@ -23,6 +23,7 @@ REGISTER2(UnaryOp, CPU, "ComplexAbs", functor::abs, complex64, complex128); #endif #if GOOGLE_CUDA REGISTER4(UnaryOp, GPU, "Abs", functor::abs, float, Eigen::half, double, int64); +REGISTER2(UnaryOp, GPU, "ComplexAbs", functor::abs, complex64, complex128); // A special GPU kernel for int32. // TODO(b/25387198): Also enable int32 in device memory. This kernel |