Home
last modified time | relevance | path

Searched defs:conv_dnums (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc49 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput()
124 const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedKernel()
Dcudnn_conv_rewriter.cc116 const ConvolutionDimensionNumbers& conv_dnums = in MatchBackwardFilter() local
Dcudnn_conv_rewriter_test.cc274 ConvolutionDimensionNumbers conv_dnums; in TEST_F() local