tract-data 0.23.4

Tiny, no-nonsense, self contained, TensorFlow and ONNX inference
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

#[macro_use]
extern crate criterion;
extern crate tract_data;

use criterion::Criterion;
use tract_data::internal::*;

fn inceptionv3_5b(c: &mut Criterion) {
    c.bench_function("inceptionv3_5b", |b| {
        b.iter_with_setup(
            || unsafe {
                vec![
                    Tensor::uninitialized_dt(DatumType::F32, &[1, 35, 35, 64]).unwrap(),
                    Tensor::uninitialized_dt(DatumType::F32, &[1, 35, 35, 64]).unwrap(),
                    Tensor::uninitialized_dt(DatumType::F32, &[1, 35, 35, 96]).unwrap(),
                    Tensor::uninitialized_dt(DatumType::F32, &[1, 35, 35, 32]).unwrap(),
                ]
            },
            |input| Tensor::stack_tensors(3, &input),
        );
    });
}

criterion_group!(benches, inceptionv3_5b);
criterion_main!(benches);