Searched defs:batch_norm (Results 1 – 11 of 11) sorted by relevance
50 bool EpsilonInRange(HloInstruction* batch_norm) { in EpsilonInRange()54 Status Visitor::HandleBatchNormInference(HloInstruction* batch_norm) { in HandleBatchNormInference()90 Status Visitor::HandleBatchNormTraining(HloInstruction* batch_norm) { in HandleBatchNormTraining()159 Status Visitor::HandleBatchNormGrad(HloInstruction* batch_norm) { in HandleBatchNormGrad()
33 batch_norm = layers.batch_norm variable
39 batch_norm = layers.batch_norm variable
34 batch_norm = layers.batch_norm variable
186 HloInstruction* batch_norm) { in HandleBatchNormTraining()327 HloInstruction* batch_norm) { in HandleBatchNormInference()432 HloInstruction* batch_norm) { in HandleBatchNormGrad()
318 batch_norm = batch_normalization variable
254 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F() local331 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F() local
343 const NodeDef* batch_norm, in FindConv2DWithBatchNorm()401 const auto batch_norm = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBatchNormAndRelu() local
269 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm() local1557 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers() local
1087 Node* batch_norm; member1093 Node* batch_norm; member1217 Node* batch_norm = conv_graph.batch_norm; in Conv2DWithBatchNormAndRelu() local
430 def batch_norm(inputs, function