diff options
author | 2017-04-11 09:53:36 -0800 | |
---|---|---|
committer | 2017-04-11 11:13:38 -0700 | |
commit | 2fe28ca92fdf4ec209a68877c9c5e76348809a8b (patch) | |
tree | 3eba125fa008bb536a811d7e7cab157cf4ea776c /tensorflow/core/kernels/cwise_ops_gpu_common.cu.h | |
parent | a2818749579f315740ea150c34fad25f87b69e13 (diff) |
Add a tf.neg GPU kernel for complex64/complex128.
Also, enable GPU tests for ops that are already supported on GPU.
Change: 152831861
Diffstat (limited to 'tensorflow/core/kernels/cwise_ops_gpu_common.cu.h')
-rw-r--r-- | tensorflow/core/kernels/cwise_ops_gpu_common.cu.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h b/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h index b8e23e2c35..6dd108f722 100644 --- a/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h +++ b/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h @@ -133,6 +133,9 @@ struct ApproximateEqual<GPUDevice, T> { #define DEFINE_UNARY6(F, T0, T1, T2, T3, T4, T5) \ DEFINE_UNARY2(F, T0, T1); \ DEFINE_UNARY4(F, T2, T3, T4, T5) +#define DEFINE_UNARY7(F, T0, T1, T2, T3, T4, T5, T6) \ + DEFINE_UNARY2(F, T0, T1); \ + DEFINE_UNARY5(F, T2, T3, T4, T5, T6) // Macros to explicitly instantiate kernels on GPU for multiple types // (T0, T1, etc.) for BinaryFunctor. |