Searched defs:tile_assignment_dims (Results 1 – 1 of 1) sorted by relevance
594 std::vector<int64> tile_assignment_dims(hlo.shape().rank()); in GatherEffectiveOutputSharding() local758 std::vector<int64> tile_assignment_dims(data_rank, 1LL); in ScatterEffectiveDataSharding() local