[−][src]Trait tract_hir::ops::expandable::Expansion

pub trait Expansion: DynClone + Debug + Send + Sync + Downcast + DynHash {
    fn name(&self) -> Cow<str>;

    fn op_families(&self) -> &'static [&'static str];

    fn wire(
        &self, 
        prefix: &str, 
        model: &mut TypedModel, 
        inputs: &[OutletId]
    ) -> TractResult<TVec<OutletId>>;

    fn rules<'r, 'p: 'r, 's: 'r>(
        &'s self, 
        s: &mut Solver<'r>, 
        inputs: &'p [TensorProxy], 
        outputs: &'p [TensorProxy]
    ) -> InferenceResult;

    fn validation(&self) -> Validation { ... }

    fn info(&self) -> TractResult<Vec<String>> { ... }

    fn nboutputs(&self) -> TractResult<usize> { ... }
}

Required methods

fn name(&self) -> Cow<str>

fn op_families(&self) -> &'static [&'static str]

fn wire(
    &self,
    prefix: &str,
    model: &mut TypedModel,
    inputs: &[OutletId]
) -> TractResult<TVec<OutletId>>

fn rules<'r, 'p: 'r, 's: 'r>(
    &'s self,
    s: &mut Solver<'r>,
    inputs: &'p [TensorProxy],
    outputs: &'p [TensorProxy]
) -> InferenceResult

Loading content...

Provided methods

fn validation(&self) -> Validation

fn info(&self) -> TractResult<Vec<String>>

fn nboutputs(&self) -> TractResult<usize>

Loading content...

Implementors

impl Expansion for Reduce[src]

Loading content...