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

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

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

fn rank_4(c: &mut Criterion) {
    c.bench_function("rank_4", |b| {
        b.iter_with_setup(
            || {
                tract_ndarray::Array4::from_shape_simple_fn((256, 35, 35, 1), || 1.0f32)
                    .permuted_axes([3, 2, 1, 0])
            },
            Tensor::from,
        );
    });
}

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