Home
last modified time | relevance | path

Searched defs:BatchNorm2d (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dbatchnorm.py51 class BatchNorm2d(_BatchNorm): class
/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_batch_norm.py502 class BatchNorm2d(torch.nn.Module): class in TestBatchNorm2d
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dbatchnorm.py378 class BatchNorm2d(_BatchNorm): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp1967 TEST_F(FunctionalTest, BatchNorm2d) { in TEST_F() argument
H A Dmodules.cpp1586 TEST_F(ModulesTest, BatchNorm2d) { in TEST_F() argument