Searched defs:module_rref (Results 1 – 2 of 2) sorted by relevance
25 def remote_device(module_rref): argument46 def get_remote_training_arg(module_rref): argument
112 def _param_rrefs(module_rref, recurse) -> List[rpc.RRef[Parameter]]: argument