Struct tract_nnef::registry::Registry[][src]

pub struct Registry {
    pub id: String,
    pub fragments: HashMap<String, FragmentDef>,
    pub primitives: HashMap<String, (Vec<Parameter>, ToTract)>,
    pub unit_element_wise_ops: Vec<(String, Box<dyn ElementWiseMiniOp>)>,
    pub element_wise_ops: Vec<(String, TypeId, FromTract, Vec<Parameter>, ToTract)>,
    pub binary_ops: Vec<(String, Box<dyn BinMiniOp>, Option<Box<dyn BinMiniOp>>)>,
    pub from_tract: HashMap<TypeId, FromTract>,
}

Fields

id: Stringfragments: HashMap<String, FragmentDef>primitives: HashMap<String, (Vec<Parameter>, ToTract)>unit_element_wise_ops: Vec<(String, Box<dyn ElementWiseMiniOp>)>element_wise_ops: Vec<(String, TypeId, FromTract, Vec<Parameter>, ToTract)>binary_ops: Vec<(String, Box<dyn BinMiniOp>, Option<Box<dyn BinMiniOp>>)>from_tract: HashMap<TypeId, FromTract>

Implementations

impl Registry[src]

pub fn new(id: impl Into<String>) -> Registry[src]

pub fn register_dumper(&mut self, id: TypeId, func: FromTract)[src]

pub fn register_primitive(
    &mut self,
    id: &str,
    decl: &[Parameter],
    func: ToTract
)
[src]

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

pub fn register_unit_element_wise(
    &mut self,
    id: impl Into<String>,
    ew: &dyn ElementWiseMiniOp
)
[src]

pub fn register_element_wise(
    &mut self,
    id: impl Into<String>,
    type_id: TypeId,
    dumper: FromTract,
    parameters: Vec<Parameter>,
    loader: ToTract
)
[src]

pub fn register_binary(&mut self, id: impl Into<String>, op: &dyn BinMiniOp)[src]

pub fn register_binary_with_flipped(
    &mut self,
    id: impl Into<String>,
    op: &dyn BinMiniOp,
    flipped: &dyn BinMiniOp
)
[src]

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

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

Auto Trait Implementations

impl !RefUnwindSafe for Registry

impl Send for Registry

impl Sync for Registry

impl Unpin for Registry

impl !UnwindSafe for Registry

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized
[src]

impl<T> Borrow<T> for T where
    T: ?Sized
[src]

impl<T> BorrowMut<T> for T where
    T: ?Sized
[src]

impl<T> Downcast for T where
    T: Any

impl<T> DowncastSync for T where
    T: Any + Send + Sync

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, 
[src]

impl<T, U> TryFrom<U> for T where
    U: Into<T>, 
[src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, 
[src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.