Struct tract_pulse::internal::IntoAst

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

Fields

framework: &'a Nnef

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

registries: Vec<String, Global>

prefix: Option<String>

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

parameters: Vec<String, Global>

results: Vec<String, Global>

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

tensors: Vec<(String, Arc<Tensor>), Global>

quantization: HashMap<String, QuantFormat, RandomState>

fragments: HashMap<String, 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 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 Into<String>) -> String

pub fn sanitize(name: impl Into<String>) -> String

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

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

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