Searched refs:MyScriptModuleWithRRefs (Results 1 – 1 of 1) sorted by relevance
225 class MyScriptModuleWithRRefs(torch.jit.ScriptModule): class282 module_with_rrefs = MyScriptModuleWithRRefs(worker_name(dst_rank))1002 m1 = MyScriptModuleWithRRefs(dst_name)1003 m2 = MyScriptModuleWithRRefs(dst_name)