Home
last modified time | relevance | path

Searched defs:group_rank (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DTypes.hpp170 int group_rank; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dset_kernels.cc109 const auto group_rank = indices.dimension(1); in CheckGroup() local
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_common.py1719 def create_dummy(store, group_rank, group_size, timeout): argument
/aosp_15_r20/external/pytorch/torch/distributed/
H A Ddistributed_c10d.py1626 group_rank, argument