Searched refs:backward_conv_dnums (Results 1 – 2 of 2) sorted by relevance
229 ConvolutionDimensionNumbers backward_conv_dnums; in MatchBackwardFilter() local230 backward_conv_dnums.set_input_batch_dimension(input_feature_dim); in MatchBackwardFilter()231 backward_conv_dnums.set_input_feature_dimension(input_batch_dim); in MatchBackwardFilter()233 backward_conv_dnums.add_input_spatial_dimensions(input_spatial_dims[i]); in MatchBackwardFilter()235 backward_conv_dnums.set_output_batch_dimension(kernel_input_feature_dim); in MatchBackwardFilter()236 backward_conv_dnums.set_output_feature_dimension(kernel_output_feature_dim); in MatchBackwardFilter()238 backward_conv_dnums.add_output_spatial_dimensions(kernel_spatial_dims[i]); in MatchBackwardFilter()245 backward_conv_dnums.set_kernel_input_feature_dimension(output_batch_dim); in MatchBackwardFilter()246 backward_conv_dnums.set_kernel_output_feature_dimension(output_feature_dim); in MatchBackwardFilter()248 backward_conv_dnums.add_kernel_spatial_dimensions(output_spatial_dims[i]); in MatchBackwardFilter()[all …]
223 ConvolutionDimensionNumbers backward_conv_dnums = in CanonicalizeBackwardFilterConvolution() local236 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardFilterConvolution()282 ConvolutionDimensionNumbers backward_conv_dnums = in CanonicalizeBackwardInputConvolution() local316 int64 dim = backward_conv_dnums.output_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()356 int64 dim = backward_conv_dnums.output_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()