|
OpenNN
Open-source neural networks library
|
Declares ConvolutionalRelu, a fused 2D convolution + ReLU layer optimized for cudnnConvolutionBiasActivationForward and CUDA Graph capture. More...
Go to the source code of this file.
Classes | |
| class | opennn::ConvolutionalRelu |
| 2D convolution + ReLU fused into a single forward op on GPU. More... | |
Namespaces | |
| namespace | opennn |
Declares ConvolutionalRelu, a fused 2D convolution + ReLU layer optimized for cudnnConvolutionBiasActivationForward and CUDA Graph capture.