Home
last modified time | relevance | path

Searched defs:fused_batch_norm_grad (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc127 int fused_batch_norm_grad = kMissingIndex; member
1789 [&](const utils::MutableNodeView& fused_batch_norm_grad) -> bool { in FindFusedBatchNormGradEx()
2148 void CopyFusedBatchNormGradAttributes(const NodeDef& fused_batch_norm_grad, in CopyFusedBatchNormGradAttributes()
2838 const NodeDef& fused_batch_norm_grad = in AddFusedBatchNormGradExNode() local
H A Dgeneric_layout_optimizer_transposer_test.cc268 auto fused_batch_norm_grad = ops::FusedBatchNormGradV2( in CreateSimpleFusedBatchNormGrad() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_ex_op_test.cc281 NodeDef fused_batch_norm_grad; in RunFusedBatchNormEx() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc880 string fused_batch_norm_grad; member