Searched refs:relu6_grad (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 83 grad = gen_nn_ops.relu6_grad(grad, y)
147 grad = gen_nn_ops.relu6_grad(grad, y)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 438 return gen_nn_ops.relu6_grad(grad, op.outputs[0])
444 return (gen_nn_ops.relu6_grad(grad, x),
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 358 csinfo_.relu6_grad = "Relu6Grad"; in MklLayoutRewritePass()
667 rinfo_.push_back({csinfo_.relu6_grad, in MklLayoutRewritePass()
668 mkl_op_registry::GetMklOpName(csinfo_.relu6_grad), in MklLayoutRewritePass()
1001 string relu6_grad; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 1995 auto* relu6_grad = context.graph_view->GetNode("relu6_grad"); in TEST_F() local
1996 ASSERT_NE(relu6_grad, nullptr); in TEST_F()
1997 TF_ASSERT_OK(unary_grad_transposer.TransposeNode(&context, relu6_grad)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | binary_ops_test.py | 180 gen_nn_ops.relu6_grad,
|