Home
last modified time | relevance | path

Searched defs:parallelize (Results 1 – 7 of 7) sorted by relevance

/external/swiftshader/third_party/marl/include/marl/
Dparallelize.h52 MARL_NO_EXPORT inline void parallelize(F0&& f0, FN&&... fn) { in parallelize() function
/external/junit/src/main/java/org/junit/experimental/
DParallelComputer.java32 private static Runner parallelize(Runner runner) { in parallelize() method in ParallelComputer
/external/pytorch/aten/src/ATen/native/
DTensorIteratorReduce.cpp127 void TensorIteratorBase::foreach_reduced_elt(loop_subiter_t loop, bool parallelize) { in foreach_reduced_elt()
/external/pytorch/test/distributed/_composable/fsdp/
Dtest_fully_shard_training.py997 def parallelize(_model: Transformer, mesh: DeviceMesh, use_seq_parallel: bool): function
1051 def parallelize(_model: Transformer, mesh: DeviceMesh, use_seq_parallel: bool): function
/external/pytorch/test/distributed/_composable/test_composability/
Dtest_2d_composability.py266 def parallelize(_model: Transformer, mesh: DeviceMesh, use_seq_parallel: bool): function
/external/pytorch/torch/testing/_internal/distributed/_tensor/
Dcommon_dtensor.py209 def parallelize( member in Transformer
/external/pytorch/torch/testing/_internal/
Dcommon_fsdp.py892 def parallelize( member in MLPStack