Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h364 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
377 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
390 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Dcuda_dnn.cc692 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
2899 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2958 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3218 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolveImpl()
3783 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3807 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3831 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3856 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3889 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
4634 const dnn::ConvolutionDescriptor& convolution_descriptor, in DeriveOutputBatchDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h349 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
362 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
375 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Drocm_dnn.cc691 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2642 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2808 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
2935 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
3439 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3456 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3473 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3491 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
4278 const dnn::ConvolutionDescriptor& convolution_descriptor, in DeriveOutputBatchDescriptor()
4315 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivationImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc465 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
503 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
541 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
580 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
618 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
656 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
698 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
740 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
782 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
824 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
[all …]
Dstream_executor_pimpl.cc297 const dnn::ConvolutionDescriptor &convolution_descriptor, in GetMIOpenConvolveAlgorithms()