Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc49 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput()
124 const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedKernel()
Dgpu_conv_rewriter.cc179 const ConvolutionDimensionNumbers& conv_dnums = in MatchBackwardFilter() local
504 auto conv_dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() local
Dgpu_conv_rewriter_test.cc276 ConvolutionDimensionNumbers conv_dnums; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.cc113 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc899 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvConvolution() local
Ddot_handler.cc143 const ConvolutionDimensionNumbers& conv_dnums = in GenNewWindow() local