Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/
H A Dmodule.h666 ModuleHolder<ModuleType> module_holder) { in register_module()
681 ModuleHolder<ModuleType> module_holder) { in replace_module()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/container/
H A Dmodulelist.h109 void push_back(const ModuleHolder<M>& module_holder) { in push_back()
236 void insert(size_t index, const ModuleHolder<M>& module_holder) { in insert()
H A Dsequential.h229 void push_back(const ModuleHolder<M>& module_holder) { in push_back()
236 void push_back(std::string name, const ModuleHolder<M>& module_holder) { in push_back()
H A Dnamed_any.h66 NamedAnyModule(std::string name, const ModuleHolder<M>& module_holder) in NamedAnyModule()
H A Dany.h243 AnyModule::AnyModule(const ModuleHolder<ModuleType>& module_holder) in AnyModule()