Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dnormalization.py228 class GroupNorm(Module): class
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dnormalization.py88 class GroupNorm(torch.nn.GroupNorm): class
/aosp_15_r20/external/executorch/backends/qualcomm/tests/
H A Dmodels.py504 class GroupNorm(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp1421 TEST_F(FunctionalTest, GroupNorm) { in TEST_F() argument
H A Dmodules.cpp1143 TEST_F(ModulesTest, GroupNorm) { in TEST_F() argument