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.cc50 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput()
126 const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedKernel()
Dgpu_conv_rewriter.cc100 const ConvolutionDimensionNumbers& conv_dnums = in MatchBackwardFilter() local
452 auto conv_dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput() local
Dgpu_conv_rewriter_test.cc277 ConvolutionDimensionNumbers conv_dnums; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.cc140 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvForDotGeneralConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc901 const auto& conv_dnums = conv.convolution_dimension_numbers(); in CreateShardedConvConvolution() local
Ddot_handler.cc156 const ConvolutionDimensionNumbers& conv_dnums = in GenNewWindow() local