Searched defs:ag_shape (Results 1 – 2 of 2) sorted by relevance
1545 int64_t channel_id, int64_t all_gather_dimension) { in GetPerGroupCollectiveOpsCreator()
4418 auto ag_shape = operand->shape(); in AllGatherShardsInternal() local