Searched refs:ModWithTwoSubmodsAndTensor (Results 1 – 3 of 3) sorted by relevance
532 from package_a.test_module import ModWithTensor, ModWithTwoSubmodsAndTensor539 mod1 = ModWithTwoSubmodsAndTensor(shared_tensor, scripted_mod_0, scripted_mod_1)586 from package_a.test_module import ModWithTensor, ModWithTwoSubmodsAndTensor593 mod1 = ModWithTwoSubmodsAndTensor(shared_tensor, scripted_mod_0, scripted_mod_1)
133 from package_a.test_module import ModWithTwoSubmodsAndTensor, SimpleTest141 module = ModWithTwoSubmodsAndTensor(
38 class ModWithTwoSubmodsAndTensor(torch.nn.Module): class