Searched refs:ModWithSubmodAndTensor (Results 1 – 2 of 2) sorted by relevance
311 ModWithSubmodAndTensor,319 ModWithSubmodAndTensor(347 from package_a.test_module import ModWithSubmodAndTensor, ModWithTensor351 ModWithSubmodAndTensor(422 from package_a.test_module import ModWithSubmodAndTensor, ModWithTensor426 ModWithSubmodAndTensor(torch.rand(1, 2, 3), scripted_mod_a)501 from package_a.test_module import ModWithSubmodAndTensor, ModWithTensor506 mod1 = ModWithSubmodAndTensor(shared_tensor, scripted_mod)507 mod2 = ModWithSubmodAndTensor(shared_tensor, scripted_mod)
28 class ModWithSubmodAndTensor(torch.nn.Module): class