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