Struct tract_pulse::internal::IntoAst

pub struct IntoAst<'a> {

    pub framework: &'a Nnef,
    pub parent: Option<&'a IntoAst<'a>>,
    pub registries: Vec<Identifier>,
    pub symbols: Vec<Symbol>,
    pub model: &'a Graph<TypedFact, Box<dyn TypedOp>>,
    pub parameters: Vec<Identifier>,
    pub results: Vec<Identifier>,
    pub mapping: HashMap<OutletId, Arc<RValue>>,
    pub tensors: HashMap<Identifier, Arc<Tensor>>,
    pub quantization: HashMap<Identifier, QuantFormat>,
    pub resources: HashMap<String, Arc<dyn Resource>>,
    pub fragments: HashMap<Identifier, FragmentDef>,
    pub body: Vec<Assignment>,
}

Fields

framework: &'a Nnef

parent: Option<&'a IntoAst<'a>>

registries: Vec<Identifier>

symbols: Vec<Symbol>

model: &'a Graph<TypedFact, Box<dyn TypedOp>>

parameters: Vec<Identifier>

results: Vec<Identifier>

mapping: HashMap<OutletId, Arc<RValue>>

tensors: HashMap<Identifier, Arc<Tensor>>

quantization: HashMap<Identifier, QuantFormat>

resources: HashMap<String, Arc<dyn Resource>>

fragments: HashMap<Identifier, FragmentDef>

body: Vec<Assignment>

Implementations

impl<'a> IntoAst<'a>

pub fn new( framework: &'a Nnef, model: &'a Graph<TypedFact, Box<dyn TypedOp>> ) -> IntoAst<'a>

pub fn ensure_symbol(&mut self, s: &Symbol) -> Result<(), Error>

pub fn into_fragment( self ) -> Result<(FragmentDef, Vec<RequiredTensorParameter>), Error>

pub fn into_proto_model(self) -> Result<ProtoModel, Error>

pub fn scoped_id(&self, name: impl AsRef<str>) -> Identifier

pub fn force_variable( &mut self, name: impl AsRef<str>, exp: &Arc<RValue> ) -> Arc<RValue>

pub fn force_variable_and_name( &mut self, name: impl Into<String>, exp: &Arc<RValue> ) -> Arc<RValue>

pub fn konst( &mut self, name: impl AsRef<str>, tensor: &Arc<Tensor> ) -> Result<Arc<RValue>, Error>

pub fn konst_variable( &mut self, name: impl AsRef<str>, tensor: &Arc<Tensor> ) -> Result<Arc<RValue>, Error>