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