Searched refs:index_to_sharding (Results 1 – 3 of 3) sorted by relevance
75 for (const auto& index_to_sharding : shardings.leaves()) { in Tuple() local76 *result.add_tuple_shardings() = index_to_sharding.second; in Tuple()
111 for (const auto& index_to_sharding : sub_shardings.leaves()) { in Tuple() local112 flattened_list.push_back(index_to_sharding.second); in Tuple()330 for (auto& index_to_sharding : result.leaves()) { in AsShapeTree() local331 index_to_sharding.second = *it++; in AsShapeTree()389 for (const auto& index_to_sharding : shape_tree.leaves()) { in ValidateTuple() local390 Status status = index_to_sharding.second.ValidateNonTuple( in ValidateTuple()391 ShapeUtil::GetSubshape(shape, index_to_sharding.first), num_devices); in ValidateTuple()395 index_to_sharding.first.ToString(), " which is ", in ValidateTuple()396 index_to_sharding.second.ToString())); in ValidateTuple()
427 for (auto& index_to_sharding : sharding_tree.leaves()) { in TEST_P() local428 index_to_sharding.second = *it++; in TEST_P()