Searched refs:fused_batch_norm_grad (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 266 grad_x, grad_scale, grad_offset, _, _ = gen_nn_ops.fused_batch_norm_grad(
316 out = gen_nn_ops.fused_batch_norm_grad(
326 ref_x, ref_scale, ref_offset, _, _ = gen_nn_ops.fused_batch_norm_grad(
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 296 NodeDef fused_batch_norm_grad; in RunFusedBatchNormEx() local
309 .Finalize(&fused_batch_norm_grad)); in RunFusedBatchNormEx()
321 {&fused_batch_norm_ex, &activation_grad, &fused_batch_norm_grad}); in RunFusedBatchNormEx()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 268 auto fused_batch_norm_grad = ops::FusedBatchNormGradV2( in CreateSimpleFusedBatchNormGrad() local
275 fused_batch_norm_grad.x_backprop); in CreateSimpleFusedBatchNormGrad()
277 fused_batch_norm_grad.scale_backprop); in CreateSimpleFusedBatchNormGrad()
279 fused_batch_norm_grad.offset_backprop); in CreateSimpleFusedBatchNormGrad()
281 fused_batch_norm_grad.reserve_space_3); in CreateSimpleFusedBatchNormGrad()
283 fused_batch_norm_grad.reserve_space_4); in CreateSimpleFusedBatchNormGrad()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 270 csinfo_.fused_batch_norm_grad = "FusedBatchNormGrad"; in MklLayoutRewritePass()
461 {csinfo_.fused_batch_norm_grad, in MklLayoutRewritePass()
462 mkl_op_registry::GetMklOpName(csinfo_.fused_batch_norm_grad), in MklLayoutRewritePass()
929 string fused_batch_norm_grad; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 879 grad_fun = gen_nn_ops.fused_batch_norm_grad
|