Searched refs:operand_window (Results 1 – 1 of 1) sorted by relevance
778 auto operand_window = window_util::MakeWindow(ones); in PartitionConvolutionTiledOutput() local780 *operand_window.mutable_dimensions(dnums.input_spatial_dimensions(i)) = in PartitionConvolutionTiledOutput()787 lhs.ReshardAsWindowedInput(operand_window, target_operand_sharding, zero); in PartitionConvolutionTiledOutput()