Searched refs:CudnnTensorDescriptor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 493 class CudnnTensorDescriptor { class
495 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor() function in stream_executor::gpu::__anonf41f84630311::CudnnTensorDescriptor
538 SE_DISALLOW_COPY_AND_ASSIGN(CudnnTensorDescriptor);
2157 const CudnnHandle& cudnn, const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionForwardAlgo()
2159 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo()
2173 const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionBackwardDataAlgo()
2176 const CudnnTensorDescriptor& output_nd, in GetCudnnConvolutionBackwardDataAlgo()
2192 const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionBackwardFilterAlgo()
2195 const CudnnTensorDescriptor& output_nd, in GetCudnnConvolutionBackwardFilterAlgo()
2211 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionForwardWorkspace()
[all …]
|