tensorflow/core/kernels/reduction_ops_all.cc


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("All")
                            .Device(DEVICE_CPU)
                            .HostMemory("reduction_indices"),
                        ReductionOp<CPUDevice, bool, functor::AllReducer>);

#if GOOGLE_CUDA
REGISTER_KERNEL_BUILDER(Name("All")
                            .Device(DEVICE_GPU)
                            .HostMemory("reduction_indices"),
                        ReductionOp<GPUDevice, bool, functor::AllReducer>);
#endif

}  // namespace tensorflow