Searched refs:data_shard_size (Results 1 – 1 of 1) sorted by relevance
3302 int64 data_shard_size = (max_windows - 1) * wd.stride() + wd.size(); in HandleSelectAndScatter() local3306 data_shard_size, i, hlo->sharding(), offset_start_in_data, pad_value, in HandleSelectAndScatter()