Home
last modified time | relevance | path

Searched defs:BatchNorm1d (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dbatchnorm.py268 class BatchNorm1d(_BatchNorm): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp1934 TEST_F(FunctionalTest, BatchNorm1d) { in TEST_F() argument
H A Dmodules.cpp1524 TEST_F(ModulesTest, BatchNorm1d) { in TEST_F() argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py121 def wrapped_with_submodule(x: torch.Tensor, batchnorm1d: torch.nn.BatchNorm1d):