Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h358 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
371 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
384 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Dcuda_dnn.cc683 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
2692 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionForwardAlgorithm()
2755 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardDataAlgorithm()
2816 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardFilterAlgorithm()
2989 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
3045 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3182 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolveImpl()
3761 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3783 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3805 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h382 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
395 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
408 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Drocm_dnn.cc758 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2849 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2945 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3103 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
3125 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3334 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3680 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3696 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3712 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3729 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dstream.h331 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveWithAlgorithm()
441 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveBackwardDataWithAlgorithm()
473 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveBackwardFilterWithAlgorithm()
Ddnn.h1148 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
1166 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
1185 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
1206 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
1249 const ConvolutionDescriptor& convolution_descriptor, in PrepareForConvolution()
1309 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
1412 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveBackwardData()
1459 const ConvolutionDescriptor& convolution_descriptor, in DoConvolveBackwardFilter()
2492 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivation()
2684 const ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
Dstream.cc451 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
482 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
513 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
545 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
576 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
607 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolve()
627 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
654 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
681 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenSeparableConvolve()
Dstream_executor_pimpl.cc283 const dnn::ConvolutionDescriptor &convolution_descriptor, in GetMIOpenConvolveAlgorithms()