Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc759 std::vector<int64_t> input_to_output_indices(output_base_shape.rank()); in PartitionConvolutionTiledOutput() local
760 input_to_output_indices[dnums.input_batch_dimension()] = in PartitionConvolutionTiledOutput()
762 input_to_output_indices[dnums.input_feature_dimension()] = in PartitionConvolutionTiledOutput()
765 input_to_output_indices[dnums.input_spatial_dimensions(i)] = in PartitionConvolutionTiledOutput()
769 output_sharding, input_to_output_indices); in PartitionConvolutionTiledOutput()