Home
last modified time | relevance | path

Searched refs:batch_norm (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc41 Status HandleBatchNormInference(HloInstruction* batch_norm) override;
42 Status HandleBatchNormTraining(HloInstruction* batch_norm) override;
43 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
57 bool EpsilonInRange(HloInstruction* batch_norm) { in EpsilonInRange() argument
58 return batch_norm->epsilon() >= 1e-5; in EpsilonInRange()
61 bool IsF32BatchNormWithFP16Inputs(HloInstruction* batch_norm) { in IsF32BatchNormWithFP16Inputs() argument
62 auto convert = batch_norm->operand(0); in IsF32BatchNormWithFP16Inputs()
69 Status Visitor::HandleBatchNormInference(HloInstruction* batch_norm) { in HandleBatchNormInference() argument
70 if (batch_norm->operand(0)->shape().element_type() != F32) { in HandleBatchNormInference()
72 << batch_norm->ToString(); in HandleBatchNormInference()
[all …]
Dir_emitter.h97 Status HandleBatchNormInference(HloInstruction* batch_norm) override;
98 Status HandleBatchNormTraining(HloInstruction* batch_norm) override;
99 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgpu_fusion.cc68 auto batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(relu_input); in matchAndRewrite() local
71 if (!batch_norm) { in matchAndRewrite()
77 batch_norm = in matchAndRewrite()
79 if (batch_norm) { in matchAndRewrite()
83 batch_norm = in matchAndRewrite()
85 if (!batch_norm) return failure(); in matchAndRewrite()
89 assert(batch_norm); in matchAndRewrite()
90 if (batch_norm.is_training()) return failure(); in matchAndRewrite()
91 if (!batch_norm.y().hasOneUse()) return failure(); in matchAndRewrite()
94 OperationState state(batch_norm.getLoc(), in matchAndRewrite()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc51 Status HandleBatchNormTraining(HloInstruction* batch_norm) override;
53 Status HandleBatchNormInference(HloInstruction* batch_norm) override;
55 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
153 HloInstruction* batch_norm) { in HandleBatchNormTraining() argument
161 added_inst->set_metadata(batch_norm->metadata()); in HandleBatchNormTraining()
172 HloInstruction* operand = batch_norm->mutable_operand(0); in HandleBatchNormTraining()
175 int64 feature_index = batch_norm->feature_index(); in HandleBatchNormTraining()
177 HloInstruction* scale = batch_norm->mutable_operand(1); in HandleBatchNormTraining()
178 HloInstruction* offset = batch_norm->mutable_operand(2); in HandleBatchNormTraining()
185 auto epsilon_literal = LiteralUtil::CreateR0(batch_norm->epsilon()); in HandleBatchNormTraining()
[all …]
Dhlo_element_type_converter_test.cc111 ::testing::Matcher<const ::xla::HloInstruction*> batch_norm = in TEST_F()
114 op::Tuple(op::Convert(op::GetTupleElement(batch_norm, 0)), in TEST_F()
115 op::Convert(op::GetTupleElement(batch_norm, 1)), in TEST_F()
116 op::Convert(op::GetTupleElement(batch_norm, 2)))); in TEST_F()
/external/tensorflow/tensorflow/python/layers/
Dnormalization.py31 batch_norm = batch_normalization variable
/external/tensorflow/tensorflow/python/ops/
Dbatch_norm_benchmark.py62 batch_norm = (tensor - mean) * math_ops.rsqrt(variance + 0.001)
64 batch_norm *= gamma
65 return batch_norm + beta
/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
207 .Input(batch_norm) in Conv2DWithBatchNormAndActivation()
211 return {graph, conv2d, batch_norm, activation}; in Conv2DWithBatchNormAndActivation()
/external/tensorflow/tensorflow/python/keras/tests/
Dgraph_util_test.py111 batch_norm = keras.layers.BatchNormalization()(inputs)
112 model = keras.models.Model(inputs, batch_norm, name="test")
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc775 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F() local
777 auto fetch = ops::Identity(s.WithOpName("fetch"), batch_norm.y); in TEST_F()
854 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, in TEST_F() local
864 return ops::Identity(fetch, ops::Relu(activate, batch_norm.y)); in TEST_F()
866 return ops::Identity(fetch, ops::Relu6(activate, batch_norm.y)); in TEST_F()
868 return ops::Identity(fetch, ops::Elu(activate, batch_norm.y)); in TEST_F()
872 fetch, ops::internal::LeakyRelu(activate, batch_norm.y, attr)); in TEST_F()
875 return ops::Identity(fetch, batch_norm.y); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlayout_optimization_to_nhwc.mlir6 %arg1: tensor<64xf32>, // batch_norm args
7 %arg2: tensor<256xf32>, // batch_norm args
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.estimator.experimental.pbtxt25 …umns\', \'activation_fn\', \'dropout\', \'input_layer_partitioner\', \'batch_norm\'], varargs=None…
Dtensorflow.estimator.-d-n-n-linear-combined-regressor.pbtxt24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
Dtensorflow.estimator.-d-n-n-estimator.pbtxt24 …out\', \'input_layer_partitioner\', \'config\', \'warm_start_from\', \'batch_norm\'], varargs=None…
Dtensorflow.estimator.-d-n-n-classifier.pbtxt24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
Dtensorflow.estimator.-d-n-n-regressor.pbtxt24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
Dtensorflow.estimator.-d-n-n-linear-combined-classifier.pbtxt24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dnormalization.py344 batch_norm = batch_normalization variable
Dnormalization_test.py694 outputs = normalization_layers.batch_norm(
748 _ = normalization_layers.batch_norm(
755 outputs2 = normalization_layers.batch_norm(
811 _ = normalization_layers.batch_norm(
815 _ = normalization_layers.batch_norm(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.-d-n-n-regressor.pbtxt25 …', \'dropout\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
Dtensorflow.estimator.-d-n-n-classifier.pbtxt25 …', \'dropout\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
Dtensorflow.estimator.-d-n-n-linear-combined-regressor.pbtxt25 …eight_column\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
Dtensorflow.estimator.-d-n-n-estimator.pbtxt25 …\', \'activation_fn\', \'dropout\', \'config\', \'warm_start_from\', \'batch_norm\'], varargs=None…
Dtensorflow.estimator.-d-n-n-linear-combined-classifier.pbtxt25 …l_vocabulary\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc886 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm() local
889 auto y = batch_norm.y; in CreateGrapplerItemWithBatchNorm()
890 auto batch_mean = batch_norm.batch_mean; in CreateGrapplerItemWithBatchNorm()
891 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithBatchNorm()
2223 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers() local
2226 auto y = batch_norm.y; in CreateGrapplerItemWithInterDeviceTransfers()
2227 auto batch_mean = batch_norm.batch_mean; in CreateGrapplerItemWithInterDeviceTransfers()
2228 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithInterDeviceTransfers()

12