Searched refs:input_to_output_indices (Results 1 – 1 of 1) sorted by relevance
757 std::vector<int64> input_to_output_indices(output_base_shape.rank()); in PartitionConvolutionTiledOutput() local758 input_to_output_indices[dnums.input_batch_dimension()] = in PartitionConvolutionTiledOutput()760 input_to_output_indices[dnums.input_feature_dimension()] = in PartitionConvolutionTiledOutput()763 input_to_output_indices[dnums.input_spatial_dimensions(i)] = in PartitionConvolutionTiledOutput()767 output_sharding, input_to_output_indices); in PartitionConvolutionTiledOutput()