/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_batchnorm_rewriter.cc | 41 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 …]
|
D | ir_emitter.h | 97 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/ |
D | gpu_fusion.cc | 68 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/ |
D | batchnorm_expander.cc | 51 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 …]
|
D | hlo_element_type_converter_test.cc | 111 ::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/ |
D | normalization.py | 31 batch_norm = batch_normalization variable
|
/external/tensorflow/tensorflow/python/ops/ |
D | batch_norm_benchmark.py | 62 batch_norm = (tensor - mean) * math_ops.rsqrt(variance + 0.001) 64 batch_norm *= gamma 65 return batch_norm + beta
|
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_ops_benchmark_test.cc | 57 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/ |
D | graph_util_test.py | 111 batch_norm = keras.layers.BatchNormalization()(inputs) 112 model = keras.models.Model(inputs, batch_norm, name="test")
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | remapper_test.cc | 775 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/ |
D | layout_optimization_to_nhwc.mlir | 6 %arg1: tensor<64xf32>, // batch_norm args 7 %arg2: tensor<256xf32>, // batch_norm args
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.estimator.experimental.pbtxt | 25 …umns\', \'activation_fn\', \'dropout\', \'input_layer_partitioner\', \'batch_norm\'], varargs=None…
|
D | tensorflow.estimator.-d-n-n-linear-combined-regressor.pbtxt | 24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
|
D | tensorflow.estimator.-d-n-n-estimator.pbtxt | 24 …out\', \'input_layer_partitioner\', \'config\', \'warm_start_from\', \'batch_norm\'], varargs=None…
|
D | tensorflow.estimator.-d-n-n-classifier.pbtxt | 24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
|
D | tensorflow.estimator.-d-n-n-regressor.pbtxt | 24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
|
D | tensorflow.estimator.-d-n-n-linear-combined-classifier.pbtxt | 24 …_partitioner\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
|
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
D | normalization.py | 344 batch_norm = batch_normalization variable
|
D | normalization_test.py | 694 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/ |
D | tensorflow.estimator.-d-n-n-regressor.pbtxt | 25 …', \'dropout\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
|
D | tensorflow.estimator.-d-n-n-classifier.pbtxt | 25 …', \'dropout\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\'], varargs=None…
|
D | tensorflow.estimator.-d-n-n-linear-combined-regressor.pbtxt | 25 …eight_column\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
|
D | tensorflow.estimator.-d-n-n-estimator.pbtxt | 25 …\', \'activation_fn\', \'dropout\', \'config\', \'warm_start_from\', \'batch_norm\'], varargs=None…
|
D | tensorflow.estimator.-d-n-n-linear-combined-classifier.pbtxt | 25 …l_vocabulary\', \'config\', \'warm_start_from\', \'loss_reduction\', \'batch_norm\', \'linear_spar…
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler_test.cc | 886 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()
|