Searched defs:index_tile_assignment_dims (Results 1 – 1 of 1) sorted by relevance
545 std::vector<int64> index_tile_assignment_dims; in GatherIndexSharding() local646 std::vector<int64> index_tile_assignment_dims; in ScatterIndexSharding() local