Skip to main content

Crate cubek_convolution

Crate cubek_convolution 

Source

Re-exports§

pub use kernels::backward_data;
pub use kernels::backward_weight;
pub use kernels::forward;
pub use launch::AcceleratedTileKind;
pub use launch::ConvAlgorithm;
pub use launch::ConvolutionArgs;
pub use launch::ConvolutionInputs;
pub use launch::Strategy;
pub use launch::launch_ref;
pub use kernels::forward::DepthwiseStrategy;
pub use kernels::forward::DepthwiseTensors;
pub use kernels::forward::DepthwiseTiling;
pub use kernels::forward::launch_depthwise;
pub use kernels::forward::DirectTensors;
pub use kernels::forward::launch_direct;

Modules§

components
definition
kernels
launch
routines