Trait tract_core::model::ModelDslConst

pub trait ModelDslConst {
    fn add_const(
        &mut self, 
        name: impl Into<String>, 
        v: impl IntoArcTensor
    ) -> TractResult<usize>;
    fn plug_const(
        &mut self, 
        inlet: InletId, 
        name: impl Into<String>, 
        v: impl IntoArcTensor
    ) -> TractResult<()>;
}

Extension to add constants to model that tolerates them.

Required methods

fn add_const(
    &mut self,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<usize>

Add a constant node to the graph.

fn plug_const(
    &mut self,
    inlet: InletId,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<()>

Add a constant node to the graph and connect its output to inlet.

Loading content...

Implementors

impl ModelDslConst for InferenceModel

fn add_const(
    &mut self,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<usize>

fn plug_const(
    &mut self,
    inlet: InletId,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<()>

impl ModelDslConst for TypedModel

fn add_const(
    &mut self,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<usize>

fn plug_const(
    &mut self,
    inlet: InletId,
    name: impl Into<String>,
    v: impl IntoArcTensor
) -> TractResult<()>

Loading content...