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/graph/ |
D | mkl_layout_pass.cc | 272 csinfo_.lrn_grad = "LRNGrad"; in MklLayoutRewritePass() 408 rinfo_.push_back({csinfo_.lrn_grad, in MklLayoutRewritePass() 409 mkl_op_registry::GetMklOpName(csinfo_.lrn_grad), in MklLayoutRewritePass() 533 wsinfo_.push_back({csinfo_.lrn, csinfo_.lrn_grad, 0, 2, 1, 3}); in MklLayoutRewritePass() 704 string lrn_grad; member 1244 if (e->dst()->type_string() == csinfo_.lrn_grad && e->dst_input() == 2 && in LrnGradRewrite()
|
/external/tensorflow/tensorflow/python/ops/ |
D | nn_grad.py | 646 gen_nn_ops.lrn_grad(grad, op.inputs[0], op.outputs[0], depth_radius, bias,
|