Searched refs:module_interface_cls (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/nn/jit/ |
| H A D | instantiator.py | 101 module_interface_cls, enable_moving_cpu_tensors_to_cuda=True argument
103 if not getattr(module_interface_cls, "__torch_script_interface__", False):
111 module_interface_cls
121 module_interface_cls
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/api/ |
| H A D | remote_module.py | 86 def _instantiate_template(module_interface_cls, enable_moving_cpu_tensors_to_cuda): argument
88 module_interface_cls, enable_moving_cpu_tensors_to_cuda
104 module_cls, args, kwargs, device, module_interface_cls argument
107 if module_interface_cls is not None:
109 return rpc.RRef(module, module_interface_cls)
483 def _init_template(self, module_interface_cls, enable_moving_cpu_tensors_to_cuda): argument
486 module_interface_cls, enable_moving_cpu_tensors_to_cuda
|