Searched defs:batch_norm (Results 1 – 9 of 9) sorted by relevance
57 bool EpsilonInRange(HloInstruction* batch_norm) { in EpsilonInRange()61 bool IsF32BatchNormWithFP16Inputs(HloInstruction* batch_norm) { in IsF32BatchNormWithFP16Inputs()69 Status Visitor::HandleBatchNormInference(HloInstruction* batch_norm) { in HandleBatchNormInference()121 Status Visitor::HandleBatchNormTraining(HloInstruction* batch_norm) { in HandleBatchNormTraining()210 Status Visitor::HandleBatchNormGrad(HloInstruction* batch_norm) { in HandleBatchNormGrad()
31 batch_norm = batch_normalization variable
68 auto batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(relu_input); in matchAndRewrite() local
153 HloInstruction* batch_norm) { in HandleBatchNormTraining()294 HloInstruction* batch_norm) { in HandleBatchNormInference()381 HloInstruction* batch_norm) { in HandleBatchNormGrad()
344 batch_norm = batch_normalization variable
57 Node* batch_norm; member63 Node* batch_norm; member203 Node* batch_norm = conv_graph.batch_norm; in Conv2DWithBatchNormAndActivation() local
775 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F() local854 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, in TEST_F() local
140 auto batch_norm = ops::FusedBatchNormV2( in CreateSimpleFusedBatchNorm() local
886 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm() local2223 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers() local