Struct tract_pulse::internal::IntoAst

pub struct IntoAst<'a> {

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

Fields

framework: &'a Nnef

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

registries: Vec<Identifier, Global>

symbols: Vec<Symbol, Global>

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

parameters: Vec<Identifier, Global>

results: Vec<Identifier, Global>

mapping: HashMap<OutletId, Arc<RValue>, RandomState>

tensors: HashMap<Identifier, Arc<Tensor>, RandomState>

quantization: HashMap<Identifier, QuantFormat, RandomState>

resources: HashMap<String, Arc<dyn Resource + 'static>, RandomState>

fragments: HashMap<Identifier, FragmentDef, RandomState>

body: Vec<Assignment, Global>

Implementations

impl<'a> IntoAst<'a>

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

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

pub fn into_fragment( self ) -> Result<(FragmentDef, Vec<RequiredTensorParameter, Global>), 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>