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::UnitStrategy;
pub use launch::UnitTiling;
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;
pub use kernels::im2col::Im2colTensors;
pub use kernels::im2col::launch_im2col;
pub use kernels::backward_weight::depthwise::DepthwiseWeightBackwardTensors;
pub use kernels::backward_weight::depthwise::launch_depthwise_weight_backward;

Modules§

components
definition
kernels
launch
routines