diff options
| author |  RJ Ryan <rjryan@google.com> | 2017-04-12 13:09:54 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-12 14:27:35 -0700 |
| commit | f2f9cdcadd3ba96c75ea7da4ef67de2a54e373cf (patch) | |
| tree | 049b348c2a65ef3e6f6282dbe5c838b0a95512a8 /tensorflow/core/kernels/cwise_op_equal_to_2.cc | |
| parent | 59b3423ee2f7fafd59d9ee4ff2efe8f4d5441485 (diff) | |
Enable tf.equal and tf.not_equal GPU kernels for complex types.
Change: 152984852
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_equal_to_2.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_equal_to_2.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cwise_op_equal_to_2.cc b/tensorflow/core/kernels/cwise_op_equal_to_2.cc index e7f4d3c07e..57e19c7202 100644 --- a/tensorflow/core/kernels/cwise_op_equal_to_2.cc +++ b/tensorflow/core/kernels/cwise_op_equal_to_2.cc @@ -25,7 +25,8 @@ namespace tensorflow { REGISTER6(BinaryOp, CPU, "Equal", functor::equal_to, int32, int64, complex64, complex128, string, bool); #if GOOGLE_CUDA -REGISTER4(BinaryOp, GPU, "Equal", functor::equal_to, int8, int16, int64, bool); +REGISTER6(BinaryOp, GPU, "Equal", functor::equal_to, int8, int16, int64, + complex64, complex128, bool); #endif // GOOGLE_CUDA #endif // !defined(__ANDROID_TYPES_SLIM__) |