Searched defs:conv_dnums (Results 1 – 6 of 6) sorted by relevance
50 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput()126 const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedKernel()
100 const ConvolutionDimensionNumbers& conv_dnums = in MatchBackwardFilter() local452 auto conv_dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() local
277 ConvolutionDimensionNumbers conv_dnums; in TEST_F() local
140 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution() local
901 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvConvolution() local
156 const ConvolutionDimensionNumbers& conv_dnums = in GenNewWindow() local