Searched refs:backward_conv_dnums (Results 1 – 2 of 2) sorted by relevance
307 ConvolutionDimensionNumbers backward_conv_dnums; in MatchBackwardFilter() local308 backward_conv_dnums.set_input_batch_dimension(input_feature_dim); in MatchBackwardFilter()309 backward_conv_dnums.set_input_feature_dimension(input_batch_dim); in MatchBackwardFilter()311 backward_conv_dnums.add_input_spatial_dimensions(input_spatial_dims[i]); in MatchBackwardFilter()313 backward_conv_dnums.set_output_batch_dimension(kernel_input_feature_dim); in MatchBackwardFilter()314 backward_conv_dnums.set_output_feature_dimension(kernel_output_feature_dim); in MatchBackwardFilter()316 backward_conv_dnums.add_output_spatial_dimensions(kernel_spatial_dims[i]); in MatchBackwardFilter()323 backward_conv_dnums.set_kernel_input_feature_dimension(output_batch_dim); in MatchBackwardFilter()324 backward_conv_dnums.set_kernel_output_feature_dimension(output_feature_dim); in MatchBackwardFilter()326 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() local320 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()364 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardInputConvolution()