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/cwise_op_sign.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_sign.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_sign.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/cwise_op_sign.cc b/tensorflow/core/kernels/cwise_op_sign.cc index ae2566cfcf..568906612a 100644 --- a/tensorflow/core/kernels/cwise_op_sign.cc +++ b/tensorflow/core/kernels/cwise_op_sign.cc @@ -19,8 +19,8 @@ namespace tensorflow { REGISTER7(UnaryOp, CPU, "Sign", functor::sign, float, double, int32, int64, complex64, Eigen::half, complex128); #if GOOGLE_CUDA -REGISTER4(UnaryOp, GPU, "Sign", functor::sign, float, Eigen::half, double, - int64); +REGISTER6(UnaryOp, GPU, "Sign", functor::sign, float, Eigen::half, double, + int64, complex64, complex128); // A special GPU kernel for int32. // TODO(b/25387198): Also enable int32 in device memory. This kernel |