Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h357 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
370 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
383 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Dcuda_dnn.cc764 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
2984 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionForwardAlgorithm()
3047 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardDataAlgorithm()
3108 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardFilterAlgorithm()
3383 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnOperationGraph()
3512 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnFusedOperationGraph()
3869 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
3925 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
4066 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveWithExecutionPlan()
4208 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetConvolveExecutionPlans()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h347 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
360 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
373 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Drocm_dnn.cc809 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2906 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2955 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3097 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
3119 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3328 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3677 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
4477 const dnn::ConvolutionDescriptor& convolution_descriptor, in DeriveOutputBatchDescriptor()
4513 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivationImpl()
4601 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h336 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveWithAlgorithm()
365 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveWithExecutionPlan()
392 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
420 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithExecutionPlan()
459 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveBackwardDataWithExecutionPlan()
487 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveBackwardDataWithAlgorithm()
519 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveBackwardFilterWithAlgorithm()
551 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveBackwardFilterWithExecutionPlan()
Dstream_executor_pimpl.h386 const dnn::ConvolutionDescriptor &convolution_descriptor, in GetFusedConvolveExecutionPlans()
Dstream_executor_pimpl.cc279 const dnn::ConvolutionDescriptor &convolution_descriptor, in GetConvolveExecutionPlans()
296 const dnn::ConvolutionDescriptor &convolution_descriptor, in GetMIOpenConvolveAlgorithms()
Ddnn.h1259 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
1280 const ConvolutionDescriptor& convolution_descriptor, in PrepareForConvolution()
2400 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivation()
2592 const ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
Dstream.cc450 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolve()
470 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
497 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
524 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenSeparableConvolve()