Searched refs:MyModuleInterface (Results 1 – 3 of 3) sorted by relevance
22 class MyModuleInterface: class43 ) = instantiator.get_arg_return_types_from_interface(MyModuleInterface)62 MyModuleInterface
355 class MyModuleInterface(torch.nn.Module): class380 return rpc.RRef(MyScriptModule(a), type_hint=MyModuleInterface)389 def script_rref_run_forward_my_script_module(rref: RRef[MyModuleInterface]) -> Tensor: argument410 rref_script_module = rpc.RRef(MyScriptModule(self.rank), MyModuleInterface)466 def use_rref_on_owner(rref: RRef[MyModuleInterface]) -> Tensor: argument820 def my_script_module_init(rank: int) -> MyModuleInterface:825 def construct_my_script_module(rank: int) -> MyModuleInterface:831 ref_script_module: RRef[MyModuleInterface], t: Tensor argument
55 class MyModuleInterface: class123 _module_interface_cls=MyModuleInterface,208 def run_forward(scripted_remote_module: MyModuleInterface): argument726 def run_forward(scripted_remote_module: MyModuleInterface): argument