Searched refs:resharded_operand_and_window (Results 1 – 2 of 2) sorted by relevance
786 auto resharded_operand_and_window = in PartitionConvolutionTiledOutput() local788 if (!resharded_operand_and_window.has_value()) { in PartitionConvolutionTiledOutput()794 resharded_operand_and_window->shard_window.dimensions( in PartitionConvolutionTiledOutput()800 create_sharded_conv(resharded_operand_and_window->sharded_input, in PartitionConvolutionTiledOutput()804 if (!resharded_operand_and_window->dynamic_slice_index_on_output in PartitionConvolutionTiledOutput()811 *resharded_operand_and_window->dynamic_slice_index_on_output, in PartitionConvolutionTiledOutput()
3115 auto resharded_operand_and_window = operand.ReshardAsWindowedInput( in HandleReduceWindow() local3117 if (!resharded_operand_and_window.has_value()) { in HandleReduceWindow()3123 resharded_operand_and_window->sharded_input->shape(), in HandleReduceWindow()3125 resharded_operand_and_window->shard_window, in HandleReduceWindow()3131 sharded_rw_shape, resharded_operand_and_window->sharded_input, in HandleReduceWindow()3132 replicated_init.hlo(), resharded_operand_and_window->shard_window, in HandleReduceWindow()3134 if (!resharded_operand_and_window->dynamic_slice_index_on_output in HandleReduceWindow()3141 *resharded_operand_and_window->dynamic_slice_index_on_output, in HandleReduceWindow()