ruda-core 0.1.2

Shared types, device contracts and utilities for Ruda.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

mod interpolation;
mod pooling;

pub use interpolation::{InterpolateMode, InterpolateOptions};
pub use pooling::calculate_pool_output_size;

mod grid_sample;
pub use grid_sample::{GridSampleOptions, GridSamplePaddingMode};

mod conv_options;
pub use conv_options::{ConvOptions, PaddedConvOptions, DeformConvOptions, ConvTransposeOptions, UnfoldOptions};
mod convolution;
pub use convolution::*;

mod unfold;
pub use unfold::{calculate_unfold_windows, calculate_unfold_shape};

mod attention;
pub use attention::AttentionModuleOptions;