tensorflow/core/kernels/cwise_op_gpu_square.cu.cc


1
2
3
4
5
6
7
8
9
10
11

#if GOOGLE_CUDA

#include "tensorflow/core/kernels/cwise_ops_gpu_common.cu.h"

namespace tensorflow {
namespace functor {
DEFINE_UNARY3(square, float, double, int64);
}  // namespace functor
}  // namespace tensorflow

#endif  // GOOGLE_CUDA