rktensor/
lib.rs

1//! rktensor has utilities for converting to flat tensor representations for
2//! use with rknpu2.
3//!
4//! The main functions are `to_tensor` and `to_tensor_with_quant`.
5
6pub use image;
7
8pub mod implementation;
9pub mod markers;
10pub mod softmax;
11
12pub use {
13    implementation::{to_tensor, to_tensor_with_quant},
14    softmax::{softmax_f16, softmax_f32},
15};