blob: 6af883e7550e4e3a6d0d31d2b482b34844f9fcd8 (
plain)
1
2
3
4
5
6
7
8
|
#include "tensorflow/core/kernels/cwise_ops_common.h"
namespace tensorflow {
REGISTER3(UnaryOp, CPU, "Inv", functor::inverse, float, double, complex64);
#if GOOGLE_CUDA
REGISTER3(UnaryOp, GPU, "Inv", functor::inverse, float, double, int64);
#endif
} // namespace tensorflow
|
