Searched refs:source_shard_hlo (Results 1 – 1 of 1) sorted by relevance
3224 auto source_shard_hlo = source.hlo(); in HandleSelectAndScatter() local3286 source_shard_hlo, source.base_shape(), source_left_halo_sizes[i], in HandleSelectAndScatter()3294 source_shard_hlo = *resharded_source; in HandleSelectAndScatter()3329 source_shard_hlo, replicated_init.hlo(), in HandleSelectAndScatter()