tract 0.22.1

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

use crate::TractResult;
use tract_hir::internal::*;

pub fn parse_costs(spec: &str) -> TractResult<Vec<(Cost, usize)>> {
    spec.split(',')
        .map(|spec| {
            let mut toks = spec.split('=');
            let name = toks.next().unwrap();
            let n = toks.next().unwrap().parse::<usize>().unwrap();
            let c = match name {
                "FMA(F32)" => Cost::FMA(f32::datum_type()),
                "Div(F32)" => Cost::Div(f32::datum_type()),
                "Buffer(F32)" => Cost::Buffer(f32::datum_type()),
                "Params(F32)" => Cost::Params(f32::datum_type()),
                _ => bail!("Unknown cost specifier {}", name),
            };
            Ok((c, n))
        })
        .collect()
}