wonnx_preprocessing/
lib.rs

1use ndarray::ArrayBase;
2use wonnx::utils::InputTensor;
3
4pub mod constant_folding;
5pub mod image;
6pub mod shape_inference;
7pub mod text;
8
9pub enum Tensor {
10    F32(ArrayBase<ndarray::OwnedRepr<f32>, ndarray::IxDyn>),
11    I32(ArrayBase<ndarray::OwnedRepr<i32>, ndarray::IxDyn>),
12    I64(ArrayBase<ndarray::OwnedRepr<i64>, ndarray::IxDyn>),
13}
14
15impl Tensor {
16    pub fn input_tensor(&self) -> InputTensor {
17        match self {
18            Tensor::F32(a) => a.as_slice().unwrap().into(),
19            Tensor::I32(a) => a.as_slice().unwrap().into(),
20            Tensor::I64(a) => a.as_slice().unwrap().into(),
21        }
22    }
23}