Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cudnn/
DConvPlaceholders.cpp205 Tensor cudnn_convolution_relu( in cudnn_convolution_relu() function
DConvShared.cpp705 Tensor cudnn_convolution_relu( in cudnn_convolution_relu() function
/external/pytorch/test/expect/
DHasDecompTest.test_has_decomposition.expect750 aten::cudnn_convolution_relu
751 aten::cudnn_convolution_relu.out
/external/pytorch/torch/
Doverrides.py187 torch.cudnn_convolution_relu,
/external/pytorch/test/nn/
Dtest_convolution.py3880 cudnn_out = torch.cudnn_convolution_relu(