Searched refs:relu_grad (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 81 grad = gen_nn_ops.relu_grad(grad, y)
145 grad = gen_nn_ops.relu_grad(grad, y)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 417 return gen_nn_ops.relu_grad(grad, op.outputs[0])
499 return (gen_nn_ops.relu_grad(grad, x),
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 356 csinfo_.relu_grad = "ReluGrad"; in MklLayoutRewritePass()
661 rinfo_.push_back({csinfo_.relu_grad, in MklLayoutRewritePass()
662 mkl_op_registry::GetMklOpName(csinfo_.relu_grad), in MklLayoutRewritePass()
999 string relu_grad; member
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | binary_ops_test.py | 174 gen_nn_ops.relu_grad,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | legalize_hlo.mlir | 778 // CHECK-LABEL: func @relu_grad(
787 func @relu_grad(%arg0: tensor<4x8xf32>, %arg1: tensor<?x?xf32>) -> tensor<4x8xf32> {
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1666 // CHECK-LABEL: func @relu_grad
1668 func @relu_grad(%gradients: tensor<4x8xf32>, %features: tensor<?x?xf32>) -> tensor<4x8xf32> {
|