Searched defs:batch_norm (Results 1 – 6 of 6) sorted by relevance
70 auto batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(relu_input); in matchAndRewrite() local
152 HloInstruction* batch_norm) { in HandleBatchNormTraining()295 HloInstruction* batch_norm) { in HandleBatchNormInference()384 HloInstruction* batch_norm) { in HandleBatchNormGrad()
65 Node* batch_norm; member71 Node* batch_norm; member214 Node* batch_norm = conv_graph.batch_norm; in Conv2DWithBatchNormAndActivation() local
893 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm() local2230 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers() local
1744 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F() local1823 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, in TEST_F() local
140 auto batch_norm = ops::FusedBatchNormV2( in CreateSimpleFusedBatchNorm() local