Searched refs:cudnn_convolution (Results 1 – 8 of 8) sorted by relevance
28 at::Tensor cudnn_convolution( in cudnn_convolution() function
240 Tensor cudnn_convolution( in cudnn_convolution() function
371 case aten::cudnn_convolution: in handleBlock()
175 KERNEL_CUDA(cudnn_convolution, lower_precision_fp) in TORCH_LIBRARY_IMPL()
746 aten::cudnn_convolution747 aten::cudnn_convolution.out
1532 output = at::cudnn_convolution( in _convolution()
220 designed specifically for binding (e.g., `cudnn_convolution`).
185 torch.cudnn_convolution,