/external/tensorflow/tensorflow/core/kernels/ |
cwise_op_gpu_tanh.cu.cc | 24 DEFINE_SIMPLE_BINARY3(tanh_grad, Eigen::half, float, double);
|
cwise_op_tanh.cc | 31 REGISTER5(SimpleBinaryOp, CPU, "TanhGrad", functor::tanh_grad, float, 34 REGISTER3(SimpleBinaryOp, GPU, "TanhGrad", functor::tanh_grad, float,
|
cwise_ops_gradients.h | 189 struct tanh_grad : base<T, Eigen::internal::scalar_tanh_gradient_op<T>> {}; struct in namespace:tensorflow::functor
|
/external/tensorflow/tensorflow/core/graph/ |
mkl_layout_pass.cc | [all...] |