Struct tract_nnef::registry::Registry

pub struct Registry {
    pub id: Identifier,
    pub docstrings: Option<Vec<String>>,
    pub aliases: Vec<Identifier>,
    pub fragments: HashMap<Identifier, FragmentDef>,
    pub primitives: HashMap<Identifier, PrimitiveDecl>,
    pub unit_element_wise_ops: Vec<(Identifier, Box<dyn ElementWiseMiniOp>)>,
    pub element_wise_ops: Vec<(Identifier, TypeId, FromTract, Vec<Parameter>, ToTract)>,
    pub binary_ops: Vec<BinOp>,
    pub from_tract: HashMap<TypeId, FromTract>,
    pub extensions: Vec<Extension>,
}

Fields

id: Identifier

docstrings: Option<Vec<String>>

aliases: Vec<Identifier>

fragments: HashMap<Identifier, FragmentDef>

primitives: HashMap<Identifier, PrimitiveDecl>

unit_element_wise_ops: Vec<(Identifier, Box<dyn ElementWiseMiniOp>)>

element_wise_ops: Vec<(Identifier, TypeId, FromTract, Vec<Parameter>, ToTract)>

binary_ops: Vec<BinOp>

from_tract: HashMap<TypeId, FromTract>

extensions: Vec<Extension>

Implementations

impl Registry

pub fn new(id: impl AsRef<str>) -> Registry

pub fn with_doc(self, docstring: impl Into<String>) -> Registry

pub fn register_dumper<O: TypedOp>(&mut self, dumper: FromTractWithOp<O>)

pub fn register_primitive( &mut self, id: impl AsRef<str>, params: &[Parameter], results: &[impl Into<Result_> + Clone], func: ToTract ) -> &mut PrimitiveDecl

pub fn register_fragment(&mut self, def: FragmentDef)

pub fn register_unit_element_wise( &mut self, id: impl AsRef<str>, ew: &dyn ElementWiseMiniOp )

pub fn register_element_wise( &mut self, id: impl AsRef<str>, type_id: TypeId, dumper: FromTract, parameters: Vec<Parameter>, loader: ToTract )

pub fn register_binary(&mut self, id: impl AsRef<str>, op: &dyn BinMiniOp)

pub fn serialize( &self, ast: &mut IntoAst<'_>, node: &TypedNode ) -> TractResult<Option<Arc<RValue>>>

pub fn deserialize( &self, builder: &mut ModelBuilder<'_>, invocation: &Invocation, dt: &[Option<DatumType>] ) -> TractResult<Option<Value>>