Expand description
rktensor has utilities for converting to flat tensor representations for use with rknpu2.
The main functions are to_tensor and to_tensor_with_quant.
Re-exports§
pub use implementation::to_tensor;pub use implementation::to_tensor_with_quant;pub use softmax::softmax_f16;pub use softmax::softmax_f32;pub use image;