Home
last modified time | relevance | path

Searched refs:cudnn_convolution_transpose (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConvPlaceholders.cpp100 at::Tensor cudnn_convolution_transpose( in cudnn_convolution_transpose() function
H A DConvShared.cpp468 Tensor cudnn_convolution_transpose( in cudnn_convolution_transpose() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dautocast.cpp372 case aten::cudnn_convolution_transpose: in handleBlock()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp176 KERNEL_CUDA(cudnn_convolution_transpose, lower_precision_fp) in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect752 aten::cudnn_convolution_transpose
753 aten::cudnn_convolution_transpose.out
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolution.cpp1541 output = at::cudnn_convolution_transpose( in _convolution()
/aosp_15_r20/external/pytorch/torch/
H A Doverrides.py186 torch.cudnn_convolution_transpose,