Searched defs:MyModuleInterface (Results 1 – 3 of 3) sorted by relevance
55 class MyModuleInterface: class208 def run_forward(scripted_remote_module: MyModuleInterface):726 def run_forward(scripted_remote_module: MyModuleInterface):
355 class MyModuleInterface(torch.nn.Module): class389 def script_rref_run_forward_my_script_module(rref: RRef[MyModuleInterface]) -> Tensor:466 def use_rref_on_owner(rref: RRef[MyModuleInterface]) -> Tensor:831 ref_script_module: RRef[MyModuleInterface], t: Tensor
22 class MyModuleInterface: class