Searched refs:replace_module (Results 1 – 4 of 4) sorted by relevance
500 std::shared_ptr<ModuleType> replace_module(516 std::shared_ptr<ModuleType> replace_module(671 std::shared_ptr<ModuleType> Module::replace_module( in replace_module() function679 std::shared_ptr<ModuleType> Module::replace_module( in replace_module() function682 return replace_module(name, module_holder.ptr()); in replace_module()
247 replace_module(key, modules_[key]); in insert()
227 replace_module(std::to_string(index), modules_[index]); in insert()
103 model.replace_module("linear", torch::nn::Linear(3, 4)), in TEST_F()115 model->l1 = model->replace_module("l1", torch::nn::Linear(5, 6)); in TEST_F()