Searched refs:CudnnTensorDescriptor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 515 class CudnnTensorDescriptor { class
517 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor() function in stream_executor::gpu::__anon040dea3a0311::CudnnTensorDescriptor
560 SE_DISALLOW_COPY_AND_ASSIGN(CudnnTensorDescriptor);
1697 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormForwardWorkspace()
1698 const CudnnTensorDescriptor& scale_offset_descriptor, in CreateBatchNormForwardWorkspace()
1720 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormBackwardWorkspace()
1721 const CudnnTensorDescriptor& scale_offset_descriptor, in CreateBatchNormBackwardWorkspace()
2412 const CudnnHandle& cudnn, const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionForwardAlgo()
2414 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo()
2451 const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionBackwardDataAlgo()
[all …]
|