Searched refs:tanh_grad (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_tanh.cc | 30 REGISTER6(SimpleBinaryOp, CPU, "TanhGrad", functor::tanh_grad, float,
33 REGISTER3(SimpleBinaryOp, GPU, "TanhGrad", functor::tanh_grad, float,
|
| D | cwise_op_gpu_tanh.cu.cc | 26 DEFINE_SIMPLE_BINARY3(tanh_grad, Eigen::half, float, double);
|
| D | cwise_ops_gradients.h | 193 struct tanh_grad : base<T, Eigen::internal::scalar_tanh_gradient_op<T>> {};
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 86 grad = gen_math_ops.tanh_grad(y, grad)
150 grad = gen_math_ops.tanh_grad(y, grad)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 361 csinfo_.tanh_grad = "TanhGrad"; in MklLayoutRewritePass()
676 rinfo_.push_back({csinfo_.tanh_grad, in MklLayoutRewritePass()
677 mkl_op_registry::GetMklOpName(csinfo_.tanh_grad), in MklLayoutRewritePass()
1004 string tanh_grad; member
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | binary_ops_test.py | 150 gen_math_ops.tanh_grad,
589 gen_math_ops.tanh_grad, lhs, rhs, expected=rhs * (1 - lhs * lhs))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 779 return gen_math_ops.tanh_grad(y, grad)
817 return grad * -2.0 * b * a, gen_math_ops.tanh_grad(a, grad)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cwise_ops_unary_test.py | 556 (gen_math_ops.tanh_grad, [-2, 2]),
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 1931 auto* tanh_grad = context.graph_view->GetNode("tanh_grad"); in TEST_F() local
1932 ASSERT_NE(tanh_grad, nullptr); in TEST_F()
1933 TF_ASSERT_OK(unary_grad_transposer.TransposeNode(&context, tanh_grad)); in TEST_F()
|