blob: ba445d1c3dd7cb5d4f6b889c71fea92d5e8591ee (
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("Conj").Device(DEVICE_CPU),
UnaryOp<CPUDevice, functor::conj<complex64>>);
#if GOOGLE_CUDA
// REGISTER_KERNEL_BUILDER(Name("Conj").Device(DEVICE_GPU),
// UnaryOp<GPUDevice, functor::conj<complex64>>);
#endif
} // namespace tensorflow
|