Home
last modified time | relevance | path

Searched refs:data_parallel (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_data_parallel.py197 out = dp.data_parallel(m, (var1, var2, float1), (0, 1))
200 out = dp.data_parallel(m, (var1, var2, float1), (1, 0))
203 out = dp.data_parallel(m, (var1, var2, float1), (0,))
224 out = dp.data_parallel(
229 out = dp.data_parallel(m, (var1, var2, float1), (0,), module_kwargs=kwarg_wrap)
236 out = dp.data_parallel(l, i, (0, 1))
282 nn.parallel.data_parallel(inner_m.to(dp_device), inp, device_ids)
350 dp.data_parallel(l, i, (0, 1))
351 self.assertRaises(AssertionError, lambda: dp.data_parallel(l, i, (0, 1)))
369 out = dp.data_parallel(l, i, dev_id)
[all …]
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dparallel.cpp174 ASSERT_THROWS_WITH(parallel::data_parallel(m, input), "Badness!"); in TEST_F()
190 auto output = parallel::data_parallel( in TEST_F()
201 auto output = parallel::data_parallel( in TEST_F()
223 auto output = parallel::data_parallel(m, input); in TEST_F()
277 auto output_dp = parallel::data_parallel(model_dp, input_dp); in TEST_F()
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A D__init__.py4 from torch.nn.parallel.data_parallel import data_parallel, DataParallel
H A Ddata_parallel.py220 def data_parallel( function
/aosp_15_r20/external/pytorch/docs/source/
H A Dnn.functional.rst213 :hidden:`data_parallel`
220 torch.nn.parallel.data_parallel
/aosp_15_r20/external/pytorch/test/onnx/model_defs/
H A Ddcgan.py53 output = nn.parallel.data_parallel(self.main, input, range(self.ngpu))
86 output = nn.parallel.data_parallel(self.main, input, range(self.ngpu))
/aosp_15_r20/external/skia/bazel/external/libjxl/
H A DBUILD.bazel31 "lib/jxl/base/data_parallel.cc",
32 "lib/jxl/base/data_parallel.h",
/aosp_15_r20/external/pytorch/
H A D.flake856 torch/distributed/_spmd/data_parallel.py: TOR901
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/parallel/
H A Ddata_parallel.h253 Tensor data_parallel(
/aosp_15_r20/external/skia/third_party/libjxl/
H A DBUILD.gn40 "../externals/libjxl/lib/jxl/base/data_parallel.cc",
/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dfaq.rst146 :func:`~torch.nn.parallel.data_parallel`. Input to each the :meth:`forward` on
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A DBUILD81 "//learning/pathways/data_parallel:__pkg__",