blob: 0d1df082f76e8d10b7fa67487d8a2fc0304e638c (
plain)
1
2
3
4
5
6
7
8
9
10
|
#include "tensorflow/core/kernels/cwise_ops_common.h"
namespace tensorflow {
REGISTER_KERNEL_BUILDER(Name("LogicalOr").Device(DEVICE_CPU),
BinaryOp<CPUDevice, functor::logical_or>);
#if GOOGLE_CUDA
REGISTER_KERNEL_BUILDER(Name("LogicalOr").Device(DEVICE_GPU),
BinaryOp<GPUDevice, functor::logical_or>);
#endif
} // namespace tensorflow
|