Home
last modified time | relevance | path

Searched refs:cudnn_convolution_add_relu (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConvPlaceholders.cpp216 Tensor cudnn_convolution_add_relu( in cudnn_convolution_add_relu() function
H A DConvShared.cpp757 Tensor cudnn_convolution_add_relu( in cudnn_convolution_add_relu() function
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect748 aten::cudnn_convolution_add_relu
749 aten::cudnn_convolution_add_relu.out
/aosp_15_r20/external/pytorch/torch/
H A Doverrides.py188 torch.cudnn_convolution_add_relu,
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_convolution.py3917 cudnn_out = torch.cudnn_convolution_add_relu(