Searched refs:MyScriptModule (Results 1 – 2 of 2) sorted by relevance
361 class MyScriptModule(torch.jit.ScriptModule): class380 return rpc.RRef(MyScriptModule(a), type_hint=MyModuleInterface)410 rref_script_module = rpc.RRef(MyScriptModule(self.rank), MyModuleInterface)422 rref_script_module = rpc.RRef(MyScriptModule(self.rank))821 return MyScriptModule(rank)905 my_local_script_module = MyScriptModule(self.rank)968 dst_name, MyScriptModule, args=(self.rank,)976 remote_script_module = rpc.remote(dst_name, MyScriptModule, args=(self.rank,))
163 class MyScriptModule(torch.nn.Module):174 my_script_module = torch.jit.script(MyScriptModule())