Skip to main content

dispatch_float

Macro dispatch_float 

Source
macro_rules! dispatch_float {
    ($dtype:expr, $op:expr, $T:ident => $body:expr) => { ... };
}
Expand description

Like [dispatch_arith] but restricted to the floating-point dtypes ONNX defines transcendental / accumulate ops over (f32, f16, bf16, f64).