Home
last modified time | relevance | path

Searched refs:parallel_apply (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Ddata_parallel.py15 from torch.nn.parallel.parallel_apply import parallel_apply
193 outputs = self.parallel_apply(replicas, inputs, module_kwargs)
209 def parallel_apply( member in DataParallel
212 return parallel_apply(
284 outputs = parallel_apply(replicas, inputs, module_kwargs, used_device_ids)
H A D__init__.py6 from torch.nn.parallel.parallel_apply import parallel_apply
H A Dparallel_apply.py24 def parallel_apply( function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/parallel/
H A Ddata_parallel.h190 std::vector<Tensor> parallel_apply(
289 auto outputs = parallel_apply(replicas, scattered_inputs, *devices);
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dparallel.cpp126 auto outputs = parallel::parallel_apply(modules, inputs); in TEST_F()
152 auto outputs = parallel::parallel_apply(modules, inputs, devices); in TEST_F()
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_data_parallel.py123 outputs = dp.parallel_apply(modules, inputs, None)
143 outputs = dp.parallel_apply(modules, inputs, None)
163 dp.parallel_apply(modules=(l1, l1), inputs=(None, None))
/aosp_15_r20/external/pytorch/docs/source/
H A Dnn.rst581 .. py:module:: torch.nn.parallel.parallel_apply