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.h339 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
352 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
365 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Dcuda_dnn.cc650 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
2631 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2689 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
2927 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolveImpl()
3348 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3372 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3396 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3421 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
4092 const dnn::ConvolutionDescriptor& convolution_descriptor, in DeriveOutputBatchDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h330 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
343 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
356 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
Drocm_dnn.cc647 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2597 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2815 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3150 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3167 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3184 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3202 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3989 const dnn::ConvolutionDescriptor& convolution_descriptor, in DeriveOutputBatchDescriptor()
4027 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivationImpl()
4115 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc205 ConvolutionDescriptor convolution_descriptor(effective_num_dimensions); in RunCudnnConvImpl() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc445 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
483 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
521 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
560 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
598 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
640 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
682 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
724 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolve()
740 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
770 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
[all …]