Home
last modified time | relevance | path

Searched +defs:torch +defs:nn (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dinstancenorm.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dbatchnorm.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Ddropout.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dadaptive.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dembedding.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dconv.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dpadding.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dnormalization.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dlinear.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dpooling.cpp3 namespace torch { namespace
4 namespace nn { namespace
H A Dtransformer.cpp5 namespace torch { namespace
6 namespace nn { namespace
H A Drnn.cpp3 namespace torch { namespace
4 namespace nn { namespace
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dupsampling.cpp7 namespace torch { namespace
8 namespace nn { namespace
H A Dbatchnorm.cpp14 namespace torch { namespace
15 namespace nn { namespace
H A Dinstancenorm.cpp4 namespace torch { namespace
5 namespace nn { namespace
H A Dpixelshuffle.cpp5 namespace torch { namespace
6 namespace nn { namespace
H A Dfold.cpp9 namespace torch { namespace
10 namespace nn { namespace
H A Ddistance.cpp5 namespace torch { namespace
6 namespace nn { namespace
H A Ddropout.cpp14 namespace torch { namespace
15 namespace nn { namespace
H A Dembedding.cpp14 namespace torch { namespace
15 namespace nn { namespace
H A Dnormalization.cpp12 namespace torch { namespace
13 namespace nn { namespace
H A D_functions.cpp6 namespace torch { namespace
7 namespace nn { namespace
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/container/
H A Dfunctional.cpp8 namespace torch { namespace
9 namespace nn { namespace
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_nn_functions.cpp32 namespace torch::autograd { namespace
98 PyObject* nn = PyModule_Create(&def); in initNNFunctions() local
/aosp_15_r20/external/pytorch/torch/csrc/api/src/python/
H A Dinit.cpp39 namespace torch { namespace
72 py::module nn = cpp.def_submodule("nn"); in init_bindings() local

12