Searched refs:CudnnFilterDescriptor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 565 class CudnnFilterDescriptor { class
567 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor() function in stream_executor::gpu::__anon040dea3a0311::CudnnFilterDescriptor
607 SE_DISALLOW_COPY_AND_ASSIGN(CudnnFilterDescriptor);
2413 const CudnnFilterDescriptor& filter, const CudnnConvolutionDescriptor& conv, in GetCudnnConvolutionForwardAlgo()
2452 const CudnnFilterDescriptor& filter, in GetCudnnConvolutionBackwardDataAlgo()
2495 const CudnnFilterDescriptor& filter, in GetCudnnConvolutionBackwardFilterAlgo()
2536 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionForwardWorkspace()
2580 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionBackwardDataWorkspace()
2626 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionBackwardFilterWorkspace()
2690 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in GetCudnnConvolutionForwardAlgorithm()
[all …]
|