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