Home
last modified time | relevance | path

Searched defs:convolution_descriptor (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h369 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
382 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
395 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
H A Dcuda_dnn.cc841 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
3057 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionForwardAlgorithm()
3120 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardDataAlgorithm()
3181 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardFilterAlgorithm()
3509 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnOperationGraph()
3636 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnFusedOperationGraph()
3962 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
4263 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
4719 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()
4822 const dnn::ConvolutionDescriptor& convolution_descriptor) { in ConvolveRunnerFromDesc()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h373 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
386 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
399 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
H A Drocm_dnn.cc812 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2909 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
3118 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3153 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()
3192 const dnn::ConvolutionDescriptor& convolution_descriptor) { in ConvolveRunnerFromDesc()
3218 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
3240 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3449 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3802 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
4526 const dnn::ConvolutionDescriptor& convolution_descriptor, in DeriveOutputBatchDescriptor()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.h343 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveWithAlgorithm()
372 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
399 const dnn::ConvolutionDescriptor &convolution_descriptor) { in ConvolveRunnerFromDesc()
419 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveRunnerFromDesc()
H A Dlazy_op_runner.h126 const ConvolutionDescriptor& convolution_descriptor; member
151 const ConvolutionDescriptor& convolution_descriptor; member
H A Ddnn.cc138 const dnn::ConvolutionDescriptor& convolution_descriptor) { in ConvolveRunnerFromDesc()
151 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetFusedConvolveRunners()
167 const dnn::ConvolutionDescriptor& convolution_descriptor, in FusedConvolveRunnerFromDesc()
H A Dstream_executor_pimpl.cc300 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()
323 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetFusedConvolveRunners()
343 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
H A Ddnn.h1307 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
1328 const ConvolutionDescriptor& convolution_descriptor, in PrepareForConvolution()
2426 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivation()
2624 const ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
H A Dstream.cc442 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolve()
463 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
490 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
517 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenSeparableConvolve()