Home
last modified time | relevance | path

Searched defs:batch_norm (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc57 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()
/external/tensorflow/tensorflow/python/layers/
Dnormalization.py31 batch_norm = batch_normalization variable
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgpu_fusion.cc68 auto batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(relu_input); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc153 HloInstruction* batch_norm) { in HandleBatchNormTraining()
294 HloInstruction* batch_norm) { in HandleBatchNormInference()
381 HloInstruction* batch_norm) { in HandleBatchNormGrad()
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dnormalization.py344 batch_norm = batch_normalization variable
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_benchmark_test.cc57 Node* batch_norm; member
63 Node* batch_norm; member
203 Node* batch_norm = conv_graph.batch_norm; in Conv2DWithBatchNormAndActivation() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc775 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F() local
854 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, in TEST_F() local
Dgeneric_layout_optimizer_transposer_test.cc140 auto batch_norm = ops::FusedBatchNormV2( in CreateSimpleFusedBatchNorm() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc886 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm() local
2223 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers() local