Searched refs:unpadded_data_shard_shape (Results 1 – 1 of 1) sorted by relevance
3220 auto unpadded_data_shard_shape = in HandleSelectAndScatter() local3244 unpadded_data_shard_shape.dimensions(i), in HandleSelectAndScatter()3251 unpadded_data_shard_shape.dimensions(i), in HandleSelectAndScatter()3252 unpadded_data_shard_shape.dimensions(i) + wd.padding_low() + in HandleSelectAndScatter()3265 unpadded_data_shard_shape.dimensions(i), wd.padding_low(), 1)) - in HandleSelectAndScatter()3274 unpadded_data_shard_shape.dimensions(i), in HandleSelectAndScatter()3275 unpadded_data_shard_shape.dimensions(i) + wd.stride() + in HandleSelectAndScatter()