Searched refs:_prepare_cpu_module (Results 1 – 1 of 1) sorted by relevance
5246 def _prepare_cpu_module( member in DistributedTest._DistTestBase5279 model, ddp_model, input, target = self._prepare_cpu_module(