blob: a89ef22b0880612d26487743571a7d1ef5ff6422 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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
|