Home
last modified time | relevance | path

Searched refs:operand_window (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc778 auto operand_window = window_util::MakeWindow(ones); in PartitionConvolutionTiledOutput() local
780 *operand_window.mutable_dimensions(dnums.input_spatial_dimensions(i)) = in PartitionConvolutionTiledOutput()
787 lhs.ReshardAsWindowedInput(operand_window, target_operand_sharding, zero); in PartitionConvolutionTiledOutput()