Home
last modified time | relevance | path

Searched full:batch_norm2d (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_conv_combos.py120 self.batch_norm2d = torch.nn.BatchNorm2d(3, affine=False)
128 x = self.batch_norm2d(x)
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Dqbatchnorm_test.py89 return torch.ops.quantized.batch_norm2d(
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dbatchnorm.py74 return torch.ops.quantized.batch_norm2d(
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml665 - quantized::batch_norm2d
1028 quantized::batch_norm2d: 1
H A Dmodel_ops.yaml419 quantized::batch_norm2d: 1
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DNormalization.cpp409 …m.impl(TORCH_SELECTIVE_NAME("quantized::batch_norm2d"), TORCH_FN(q_batch_norm2d_impl<false>)); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/_export/passes/
H A Dreplace_quantized_ops_with_standard_ops_pass.py488 "batch_norm2d.default": _transform_batch_norm,
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dfuse_ops.py317 exir_ops.edge.quantized.batch_norm2d.default,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A Dlibrary.cpp51 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::batch_norm2d(Tensor qx, Tensor? weight, Tensor? bias, Ten… in TORCH_LIBRARY()
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py2752 qy = torch.ops.quantized.batch_norm2d(