Searched refs:lrn_grad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lrn_ops_test.py | 118 expected = gen_nn_ops.lrn_grad(out_grads, in_image, out_image,
121 actual = gen_nn_ops.lrn_grad(out_grads, in_image, out_image,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 283 csinfo_.lrn_grad = "LRNGrad"; in MklLayoutRewritePass()
509 rinfo_.push_back({csinfo_.lrn_grad, in MklLayoutRewritePass()
510 mkl_op_registry::GetMklOpName(csinfo_.lrn_grad), in MklLayoutRewritePass()
701 wsinfo_.push_back({csinfo_.lrn, csinfo_.lrn_grad, 0, 2, 1, 3}); in MklLayoutRewritePass()
942 string lrn_grad; member
1616 if (e->dst()->type_string() == csinfo_.lrn_grad && e->dst_input() == 2 && in LrnGradRewrite()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 655 gen_nn_ops.lrn_grad(grad, op.inputs[0], op.outputs[0], depth_radius, bias,
|