Searched refs:fused_batch_norm_grad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 229 grad_x, grad_scale, grad_offset, _, _ = gen_nn_ops.fused_batch_norm_grad(
279 out = gen_nn_ops.fused_batch_norm_grad(
289 ref_x, ref_scale, ref_offset, _, _ = gen_nn_ops.fused_batch_norm_grad(
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 266 csinfo_.fused_batch_norm_grad = "FusedBatchNormGrad"; in MklLayoutRewritePass()
398 {csinfo_.fused_batch_norm_grad, in MklLayoutRewritePass()
399 mkl_op_registry::GetMklOpName(csinfo_.fused_batch_norm_grad), in MklLayoutRewritePass()
698 string fused_batch_norm_grad; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 866 if use_v2 else gen_nn_ops.fused_batch_norm_grad)
|