Searched refs:CudnnConvolutionDescriptor (Results 1 – 1 of 1) sorted by relevance
680 class CudnnConvolutionDescriptor { class682 CudnnConvolutionDescriptor( in CudnnConvolutionDescriptor() function in stream_executor::gpu::__anon040dea3a0311::CudnnConvolutionDescriptor740 SE_DISALLOW_COPY_AND_ASSIGN(CudnnConvolutionDescriptor);745 static bool IsTensorMathOpSet(const CudnnConvolutionDescriptor& conv) { in IsTensorMathOpSet()2413 const CudnnFilterDescriptor& filter, const CudnnConvolutionDescriptor& conv, in GetCudnnConvolutionForwardAlgo()2453 const CudnnConvolutionDescriptor& conv, in GetCudnnConvolutionBackwardDataAlgo()2496 const CudnnConvolutionDescriptor& conv, in GetCudnnConvolutionBackwardFilterAlgo()2537 const CudnnConvolutionDescriptor& conv, in AllocateCudnnConvolutionForwardWorkspace()2581 const CudnnConvolutionDescriptor& conv, in AllocateCudnnConvolutionBackwardDataWorkspace()2627 const CudnnConvolutionDescriptor& conv, in AllocateCudnnConvolutionBackwardFilterWorkspace()[all …]