Searched refs:CudnnFilterDescriptor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 543 class CudnnFilterDescriptor { class
545 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor() function in stream_executor::gpu::__anonf41f84630311::CudnnFilterDescriptor
585 SE_DISALLOW_COPY_AND_ASSIGN(CudnnFilterDescriptor);
2158 const CudnnFilterDescriptor& filter, const CudnnConvolutionDescriptor& conv, in GetCudnnConvolutionForwardAlgo()
2174 const CudnnFilterDescriptor& filter, in GetCudnnConvolutionBackwardDataAlgo()
2193 const CudnnFilterDescriptor& filter, in GetCudnnConvolutionBackwardFilterAlgo()
2211 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionForwardWorkspace()
2254 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionBackwardDataWorkspace()
2299 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionBackwardFilterWorkspace()
2344 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in GetCudnnConvolutionForwardAlgorithm()
[all …]
|