| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | batchnorm_expander.cc | 152 HloInstruction* batch_norm) { in HandleBatchNormTraining()
295 HloInstruction* batch_norm) { in HandleBatchNormInference()
384 HloInstruction* batch_norm) { in HandleBatchNormGrad()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/modules/ |
| H A D | fused.py | 169 def __init__(self, batch_norm, relu): argument
181 def __init__(self, batch_norm, relu): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_benchmark_test.cc | 65 Node* batch_norm; member
71 Node* batch_norm; member
214 Node* batch_norm = conv_graph.batch_norm; in Conv2DWithBatchNormAndActivation() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | batchnorm.h | 13 inline Tensor batch_norm( in batch_norm() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | gpu_fusion.cc | 70 auto batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(relu_input); in matchAndRewrite() local
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset14.py | 69 def batch_norm( function
|
| H A D | symbolic_opset9.py | 2626 def batch_norm( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Batchnorm.cpp | 74 Tensor batch_norm( in batch_norm() function
|
| /aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
| H A D | pt_engine.py | 33 def batch_norm(self, data, mean, var, training): member in TorchTensorEngine
|
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | torchaudio_models.py | 187 batch_norm=True, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | virtual_scheduler_test.cc | 893 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm() local
2230 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 623 std::string batch_norm = R"( in quant_fusion_pattern_and_replacements() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | remapper_test.cc | 1744 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F() local
1823 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, in TEST_F() local
|
| H A D | generic_layout_optimizer_transposer_test.cc | 140 auto batch_norm = ops::FusedBatchNormV2( in CreateSimpleFusedBatchNorm() local
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 897 def batch_norm( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 668 Tensor batch_norm( in batch_norm() function
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 2781 def batch_norm( function
|