1 2 3 4 5 6 7 8 9 10 11 12 13
mod data_formats; mod reduce; pub use self::data_formats::{BaseDataShape, DataFormat, DataShape}; pub use self::reduce::{Reduce, Reducer}; pub use crate::internal::*; element_wise!(sigmoid, Sigmoid, [f32] => |_, xs| { (tract_linalg::ops().sigmoid_f32)().run(xs) }; cost: |dt| {tvec!((Cost::FMA(dt), 11), (Cost::Div(dt), 1))} );