Searched refs:data_tile_assignment_dims (Results 1 – 1 of 1) sorted by relevance
780 std::vector<int64_t> data_tile_assignment_dims; in ScatterDataSharding() local783 data_tile_assignment_dims.reserve(rank); in ScatterDataSharding()786 data_tile_assignment_dims.push_back(1); in ScatterDataSharding()788 data_tile_assignment_dims.push_back( in ScatterDataSharding()801 data_tile_assignment_dims.push_back( in ScatterDataSharding()806 new_tile_assignment.Reshape(data_tile_assignment_dims); in ScatterDataSharding()