Searched refs:data_shard_hlo (Results 1 – 1 of 1) sorted by relevance
3225 auto data_shard_hlo = operand.hlo(); in HandleSelectAndScatter() local3304 data_shard_hlo, operand.base_shape(), data_left_halo_sizes[i], in HandleSelectAndScatter()3311 data_shard_hlo = *resharded_data; in HandleSelectAndScatter()3328 data_shard_hlo->shape(), data_shard_hlo, select, window_on_shard, in HandleSelectAndScatter()