init_from_module_rref (reference) in projects: aosp_15_r20

Searched refs:init_from_module_rref (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/nn/api/
H A D	remote_module.py	`509 def init_from_module_rref( member in _RemoteModule`
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/nn/api/
H A D	remote_module_test.py	`555 RemoteModule.init_from_module_rref,`