Searched refs:new_dnums (Results 1 – 4 of 4) sorted by relevance
117 ConvolutionDimensionNumbers new_dnums; in Run() local118 new_dnums.set_input_batch_dimension(0); in Run()119 new_dnums.set_output_batch_dimension(0); in Run()121 new_dnums.add_input_spatial_dimensions(i + 1); in Run()122 new_dnums.add_kernel_spatial_dimensions(i); in Run()123 new_dnums.add_output_spatial_dimensions(i + 1); in Run()125 new_dnums.set_input_feature_dimension(num_dims - 1); in Run()126 new_dnums.set_output_feature_dimension(num_dims - 1); in Run()127 new_dnums.set_kernel_input_feature_dimension(num_dims - 2); in Run()128 new_dnums.set_kernel_output_feature_dimension(num_dims - 1); in Run()[all …]
158 ConvolutionDimensionNumbers new_dnums = dnums; in FoldTransposeIntoConvolution() local170 new_dnums.set_input_batch_dimension( in FoldTransposeIntoConvolution()172 new_dnums.set_input_feature_dimension( in FoldTransposeIntoConvolution()175 *new_dnums.mutable_input_spatial_dimensions()) { in FoldTransposeIntoConvolution()193 new_dnums.set_kernel_input_feature_dimension( in FoldTransposeIntoConvolution()195 new_dnums.set_kernel_output_feature_dimension( in FoldTransposeIntoConvolution()198 *new_dnums.mutable_kernel_spatial_dimensions()) { in FoldTransposeIntoConvolution()208 convolution.batch_group_count(), convolution.window(), new_dnums, in FoldTransposeIntoConvolution()
1763 DotDimensionNumbers new_dnums; in RemoveDegenerateDimensionFromDot() local1767 new_dnums.add_lhs_batch_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()1773 new_dnums.add_lhs_contracting_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()1780 new_dnums.add_rhs_batch_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()1786 new_dnums.add_rhs_contracting_dimensions(new_dim); in RemoveDegenerateDimensionFromDot()1804 MakeDotHlo(new_lhs, new_rhs, new_dnums, dot->precision_config(), in RemoveDegenerateDimensionFromDot()4721 DotDimensionNumbers new_dnums = dnums; in HandleReduce() local4722 new_dnums.clear_lhs_batch_dimensions(); in HandleReduce()4723 new_dnums.clear_rhs_batch_dimensions(); in HandleReduce()4728 new_dnums.add_rhs_contracting_dimensions( in HandleReduce()[all …]
306 ConvolutionDimensionNumbers new_dnums; in GenNewConvDNums() local307 new_dnums.set_input_batch_dimension(input_batch_dimension); in GenNewConvDNums()308 new_dnums.set_input_feature_dimension(input_feature_dimension); in GenNewConvDNums()310 new_dnums.add_input_spatial_dimensions(dim); in GenNewConvDNums()312 new_dnums.set_kernel_input_feature_dimension(kernel_input_feature_dimension); in GenNewConvDNums()313 new_dnums.set_kernel_output_feature_dimension( in GenNewConvDNums()316 new_dnums.add_kernel_spatial_dimensions(dim); in GenNewConvDNums()318 new_dnums.set_output_batch_dimension(output_batch_dimension); in GenNewConvDNums()319 new_dnums.set_output_feature_dimension(output_feature_dimension); in GenNewConvDNums()321 new_dnums.add_output_spatial_dimensions(dim); in GenNewConvDNums()[all …]