Searched refs:input_to_output_indices (Results 1 – 1 of 1) sorted by relevance
759 std::vector<int64_t> input_to_output_indices(output_base_shape.rank()); in PartitionConvolutionTiledOutput() local760 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()