diff options
Diffstat (limited to 'tensorflow/core/kernels/reduction_ops_any.cc')
-rw-r--r-- | tensorflow/core/kernels/reduction_ops_any.cc | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/reduction_ops_any.cc b/tensorflow/core/kernels/reduction_ops_any.cc new file mode 100644 index 0000000000..a89ef22b08 --- /dev/null +++ b/tensorflow/core/kernels/reduction_ops_any.cc @@ -0,0 +1,17 @@ +#include "tensorflow/core/kernels/reduction_ops_common.h" + +namespace tensorflow { + +REGISTER_KERNEL_BUILDER(Name("Any") + .Device(DEVICE_CPU) + .HostMemory("reduction_indices"), + ReductionOp<CPUDevice, bool, functor::AnyReducer>); + +#if GOOGLE_CUDA +REGISTER_KERNEL_BUILDER(Name("Any") + .Device(DEVICE_GPU) + .HostMemory("reduction_indices"), + ReductionOp<GPUDevice, bool, functor::AnyReducer>); +#endif + +} // namespace tensorflow |