Searched refs:tanh_grad (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_tanh.cc | 31 REGISTER6(SimpleBinaryOp, CPU, "TanhGrad", functor::tanh_grad, float,
34 REGISTER3(SimpleBinaryOp, GPU, "TanhGrad", functor::tanh_grad, float,
|
| D | cwise_op_gpu_tanh.cu.cc | 27 DEFINE_SIMPLE_BINARY3(tanh_grad, Eigen::half, float, double);
|
| D | cwise_ops_gradients.h | 188 struct tanh_grad : base<T, Eigen::internal::scalar_tanh_gradient_op<T>> {};
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 82 grad = gen_math_ops.tanh_grad(y, grad)
146 grad = gen_math_ops.tanh_grad(y, grad)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 367 csinfo_.tanh_grad = "TanhGrad"; in MklLayoutRewritePass()
694 rinfo_.push_back({csinfo_.tanh_grad, in MklLayoutRewritePass()
695 mkl_op_registry::GetMklOpName(csinfo_.tanh_grad), in MklLayoutRewritePass()
958 string tanh_grad; member
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | binary_ops_test.py | 145 gen_math_ops.tanh_grad,
578 gen_math_ops.tanh_grad, lhs, rhs, expected=rhs * (1 - lhs * lhs))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 780 return gen_math_ops.tanh_grad(y, grad)
818 return grad * -2.0 * b * a, gen_math_ops.tanh_grad(a, grad)
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | cwise_ops_unary_test.py | 615 (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()
|