Home
last modified time | relevance | path

Searched defs:num_of_instances (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/iter/
H A Dsharding.py52 self, num_of_instances, instance_id, sharding_group=SHARDING_PRIORITIES.DEFAULT argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_datapipe.py3326 def apply_sharding(self, num_of_instances, instance_id): argument