ruDNN 0.1.36

Neural network operators for Ruda.
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
use ruda_kernel::dsl as kernel_dsl;
use ruda_kernel::dsl::prelude::*;
use ruda_kernel::library::FastDivmod;

use crate::convolution::components::ConvolutionOperation;

#[derive(RudaType, RudaLaunch, Clone)]
pub struct RuntimeArgs {
    pub shape_k: u32,
    pub channels: u32,
    pub padded_channels: FastDivmod<u32>,
    #[ruda(comptime)]
    pub operation: ConvolutionOperation,
}