Searched refs:sharded_select_and_scatter (Results 1 – 1 of 1) sorted by relevance
3326 auto sharded_select_and_scatter = in HandleSelectAndScatter() local3333 if (ShapeUtil::Compatible(sharded_select_and_scatter->shape(), in HandleSelectAndScatter()3335 return sharded_select_and_scatter; in HandleSelectAndScatter()3361 shard_shape, sharded_select_and_scatter, slice_offsets, in HandleSelectAndScatter()