Searched refs:ModuleHolder (Results 1 – 13 of 13) sorted by relevance
26 class ModuleHolder : torch::detail::ModuleHolderIndicator {43 ModuleHolder() : impl_(default_construct()) { in ModuleHolder() function54 /* implicit */ ModuleHolder(std::nullptr_t) : impl_(nullptr) {} in ModuleHolder() function64 explicit ModuleHolder(Head&& head, Tail&&... tail) in ModuleHolder() function71 /* implicit */ ModuleHolder(std::shared_ptr<Contained> module) in ModuleHolder() function157 const nn::ModuleHolder<ModuleType>& module) {165 const nn::ModuleHolder<ModuleType>& module) {173 nn::ModuleHolder<ModuleType>& module) {193 class Name : public torch::nn::ModuleHolder<ImplType> { /* NOLINT */ \195 using torch::nn::ModuleHolder<ImplType>::ModuleHolder; \
486 ModuleHolder<ModuleType> module_holder);518 ModuleHolder<ModuleType> module_holder);666 ModuleHolder<ModuleType> module_holder) { in register_module()681 ModuleHolder<ModuleType> module_holder) { in replace_module()
60 class Embedding : public torch::nn::ModuleHolder<EmbeddingImpl> {62 using torch::nn::ModuleHolder<EmbeddingImpl>::ModuleHolder;139 class EmbeddingBag : public torch::nn::ModuleHolder<EmbeddingBagImpl> {141 using torch::nn::ModuleHolder<EmbeddingBagImpl>::ModuleHolder;
229 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()375 class Sequential : public torch::nn::ModuleHolder<SequentialImpl> {377 using torch::nn::ModuleHolder<SequentialImpl>::ModuleHolder;379 Sequential() : ModuleHolder() {} in Sequential()385 : ModuleHolder(std::make_shared<SequentialImpl>(named_modules)) {} in Sequential()
109 void push_back(const ModuleHolder<M>& module_holder) { in push_back()236 void insert(size_t index, const ModuleHolder<M>& module_holder) { in insert()
126 explicit AnyModule(const ModuleHolder<ModuleType>& module_holder);243 AnyModule::AnyModule(const ModuleHolder<ModuleType>& module_holder) in AnyModule()
66 NamedAnyModule(std::string name, const ModuleHolder<M>& module_holder) in NamedAnyModule()
55 struct M : torch::nn::ModuleHolder<MImpl> { in TEST_F()56 using torch::nn::ModuleHolder<MImpl>::ModuleHolder; in TEST_F()57 using torch::nn::ModuleHolder<MImpl>::get; in TEST_F()
25 template <typename ModuleHolder>26 ModuleHolder add(27 ModuleHolder module_holder,
312 struct M : torch::nn::ModuleHolder<MImpl> { in TEST_F()313 using torch::nn::ModuleHolder<MImpl>::ModuleHolder; in TEST_F()314 using torch::nn::ModuleHolder<MImpl>::get; in TEST_F()
80 struct M : torch::nn::ModuleHolder<MImpl> { in TEST_F()81 using torch::nn::ModuleHolder<MImpl>::ModuleHolder; in TEST_F()82 using torch::nn::ModuleHolder<MImpl>::get; in TEST_F()
405 torch::nn::ModuleHolder<Impl>& target, in copyParameters()407 const torch::nn::ModuleHolder<Impl>& source, in copyParameters()
168 std::vector<ModuleHolder<ModuleType>> replicate( in replicate()169 const ModuleHolder<ModuleType>& module, in replicate()172 return std::vector<ModuleHolder<ModuleType>>(ptrs.begin(), ptrs.end()); in replicate()