Searched refs:NestedModule (Results 1 – 2 of 2) sorted by relevance
232 class NestedModule(torch.nn.Module): class in NestedModuleTest249 self.module = NestedModuleTest.NestedModule(BasicModule())252 NestedModuleTest.NestedModule(BasicModule())263 NestedModuleTest.NestedModule(lowered_module)
527 struct NestedModule : public Cloneable<NestedModule> { in TEST_F() struct528 NestedModule() { in TEST_F() function538 auto a = std::make_shared<NestedModule>(); in TEST_F()545 auto b = std::dynamic_pointer_cast<NestedModule>(a->clone()); in TEST_F()