cubek-convolution 0.4.0-pre.1

CubeK: Convolution Kernels
Documentation
1
2
3
4
5
6
7
8
9
10
mod base;
mod inputs;
mod strategy;
pub(crate) mod unit;
mod unit_strategy;

pub use base::launch_ref;
pub use inputs::{ConvolutionArgs, ConvolutionInputs};
pub use strategy::{AcceleratedTileKind, ConvAlgorithm, Strategy};
pub use unit_strategy::{UnitStrategy, UnitTiling};