blob: 3a19b2e94f671e67bd23b9d35c2b4747fbcb7475 (
plain)
1
2
3
4
5
6
7
8
9
|
#include "tensorflow/core/kernels/cwise_ops_common.h"
namespace tensorflow {
REGISTER5(UnaryOp, CPU, "Neg", functor::neg, float, double, int32, complex64,
int64);
#if GOOGLE_CUDA
REGISTER4(UnaryOp, GPU, "Neg", functor::neg, float, double, int32, int64);
#endif
} // namespace tensorflow
|