[−][src]Trait tract_core::ops::Op

pub trait Op: Debug + Clone + Send + Sync + 'static + Downcast + StatefullOp {
    fn name(&self) -> Cow<str>;
    fn as_typed(&self) -> Option<&dyn TypedOp>;

    fn incorporate(
        &self, 
        _model: &InferenceModel, 
        _node: &InferenceNode
    ) -> TractResult<Option<InferenceModelPatch>> { ... }
    fn declutter(
        &self, 
        _model: &TypedModel, 
        _node: &TypedNode
    ) -> TractResult<Option<TypedModelPatch>> { ... }
    fn codegen(
        &self, 
        _model: &TypedModel, 
        _node: &TypedNode
    ) -> TractResult<Option<TypedModelPatch>> { ... }
    fn fuse(
        &self, 
        _model: &TypedModel, 
        _node: &TypedNode
    ) -> TractResult<Option<TypedModelPatch>> { ... }
    fn cost(
        &self, 
        _inputs: &[&TypedTensorInfo]
    ) -> TractResult<TVec<(Cost, TDim)>> { ... }
    fn nested_models(&self) -> Vec<(Cow<str>, &dyn Model)> { ... }
    fn validation(&self) -> Validation { ... }
    fn axes_info(
        &self, 
        _model: &TypedModel, 
        _node: &TypedNode
    ) -> TractResult<AxesInfo> { ... }
    fn same_as(&self, _other: &dyn Op) -> bool { ... }
    fn info(&self) -> TractResult<Vec<String>> { ... }
    fn is_canonic(&self) -> bool { ... }
}

A base operation

Required methods

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

`fn as_typed(&self) -> Option<&dyn TypedOp>`

Loading content...

Provided methods

`fn incorporate( &self, _model: &InferenceModel, _node: &InferenceNode ) -> TractResult<Option<InferenceModelPatch>>`

Early pass on inference model, after analyse, but before translation to typed network. Meant to deal with some framework idiosyncrasies that manifest with temporaries nodes that can run some form of inference but require refactoring the network before it can be evaluated.

Called after succesful analyse, but before translating to typed model.

`fn declutter( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<Option<TypedModelPatch>>`

Declutter the op to the tract_core operator set as much as possible.

`fn codegen( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<Option<TypedModelPatch>>`

Translate the op into the most efficient form possible for execution.

This transformation is supposed to be final, no more pass are expected to be run on the codegen networks.

`fn fuse( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<Option<TypedModelPatch>>`

Fuse op after codegen to deal with local optimisations.

`fn cost(&self, _inputs: &[&TypedTensorInfo]) -> TractResult<TVec<(Cost, TDim)>>`

Computes a cost hint of the operation.

Each pair is a type of operation and a number per call on eval.

`fn nested_models(&self) -> Vec<(Cow<str>, &dyn Model)>`

Nested models, with label (for audit).

`fn validation(&self) -> Validation`

The kind of accuracy check that should be performed on operation when testing them.

`fn axes_info( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<AxesInfo>`

`fn same_as(&self, _other: &dyn Op) -> bool`

Compare two ops.

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

Short (one-line) strings giving hints on internal implementation or important configuration details to be displayed in dumps.

`fn is_canonic(&self) -> bool`

Loading content...

Methods

`impl dyn Op`

`pub fn is<__T: Op>(&self) -> bool`

Returns true if the trait object wraps an object of type __T.

`pub fn downcast<T: Op>(self: Box<Self>) -> Result<Box<T>, Box<Self>>`

Returns a boxed object from a boxed trait object if the underlying object is of type __T. Returns the original boxed trait if it isn't.

`pub fn downcast_ref<T: Op>(&self) -> Option<&T>`

Returns a reference to the object within the trait object if it is of type __T, or None if it isn't.

`pub fn downcast_mut<T: Op>(&mut self) -> Option<&mut T>`

Returns a mutable reference to the object within the trait object if it is of type __T, or None if it isn't.

Trait Implementations

`impl AsMut<dyn Op + 'static> for dyn InferenceOp`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsMut<dyn Op + 'static> for Box<dyn InferenceOp>`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsMut<dyn Op + 'static> for dyn TypedOp`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsMut<dyn Op + 'static> for Box<dyn TypedOp>`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsRef<dyn Op + 'static> for dyn InferenceOp`[src]

`fn as_ref(&self) -> &dyn Op`[src]

`impl AsRef<dyn Op + 'static> for Box<dyn InferenceOp>`[src]

`fn as_ref(&self) -> &dyn Op`[src]

`impl AsRef<dyn Op + 'static> for dyn TypedOp`[src]

`fn as_ref(&self) -> &dyn Op`[src]

`impl AsRef<dyn Op + 'static> for Box<dyn TypedOp>`[src]

`fn as_ref(&self) -> &dyn Op`[src]

Implementors

[−][src]Trait tract_core::ops::Op

Required methods

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

fn as_typed(&self) -> Option<&dyn TypedOp>

Provided methods

fn incorporate( &self, _model: &InferenceModel, _node: &InferenceNode) -> TractResult<Option<InferenceModelPatch>>

fn declutter( &self, _model: &TypedModel, _node: &TypedNode) -> TractResult<Option<TypedModelPatch>>

fn codegen( &self, _model: &TypedModel, _node: &TypedNode) -> TractResult<Option<TypedModelPatch>>

fn fuse( &self, _model: &TypedModel, _node: &TypedNode) -> TractResult<Option<TypedModelPatch>>

fn cost(&self, _inputs: &[&TypedTensorInfo]) -> TractResult<TVec<(Cost, TDim)>>

fn nested_models(&self) -> Vec<(Cow<str>, &dyn Model)>

fn validation(&self) -> Validation

fn axes_info( &self, _model: &TypedModel, _node: &TypedNode) -> TractResult<AxesInfo>

fn same_as(&self, _other: &dyn Op) -> bool

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

fn is_canonic(&self) -> bool

Methods

impl dyn Op

pub fn is<__T: Op>(&self) -> bool

pub fn downcast<__T: Op>(self: Box<Self>) -> Result<Box<__T>, Box<Self>>

pub fn downcast_ref<__T: Op>(&self) -> Option<&__T>

pub fn downcast_mut<__T: Op>(&mut self) -> Option<&mut __T>

Trait Implementations

impl AsMut<dyn Op + 'static> for dyn InferenceOp[src]

fn as_mut(&mut self) -> &mut dyn Op[src]

impl AsMut<dyn Op + 'static> for Box<dyn InferenceOp>[src]

fn as_mut(&mut self) -> &mut dyn Op[src]

impl AsMut<dyn Op + 'static> for dyn TypedOp[src]

fn as_mut(&mut self) -> &mut dyn Op[src]

impl AsMut<dyn Op + 'static> for Box<dyn TypedOp>[src]

fn as_mut(&mut self) -> &mut dyn Op[src]

impl AsRef<dyn Op + 'static> for dyn InferenceOp[src]

fn as_ref(&self) -> &dyn Op[src]

impl AsRef<dyn Op + 'static> for Box<dyn InferenceOp>[src]

fn as_ref(&self) -> &dyn Op[src]

impl AsRef<dyn Op + 'static> for dyn TypedOp[src]

fn as_ref(&self) -> &dyn Op[src]

impl AsRef<dyn Op + 'static> for Box<dyn TypedOp>[src]

fn as_ref(&self) -> &dyn Op[src]

Implementors

impl Op for AddDims[src]

fn name(&self) -> Cow<str>[src]

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

fn axes_info( &self, _model: &TypedModel, node: &TypedNode) -> TractResult<AxesInfo>[src]

fn is_canonic(&self) -> bool[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for Concat[src]

fn name(&self) -> Cow<str>[src]

fn declutter( &self, model: &TypedModel, node: &TypedNode) -> TractResult<Option<TypedModelPatch>>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for ConstantLike[src]

fn name(&self) -> Cow<str>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for ConstantOfShape[src]

fn name(&self) -> Cow<str>[src]

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

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for Crop[src]

fn name(&self) -> Cow<str>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for EyeLike[src]

fn name(&self) -> Cow<str>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for Flatten[src]

fn name(&self) -> Cow<str>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for Gather[src]

fn name(&self) -> Cow<str>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for IntoShape[src]

fn name(&self) -> Cow<str>[src]

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

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for MultiBroadcastTo[src]

fn name(&self) -> Cow<str>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for Pad[src]

fn name(&self) -> Cow<str>[src]

fn as_typed(&self) -> Option<&dyn TypedOp>[src]

impl Op for PermuteAxes[src]

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

`fn as_typed(&self) -> Option<&dyn TypedOp>`

`fn incorporate( &self, _model: &InferenceModel, _node: &InferenceNode ) -> TractResult<Option<InferenceModelPatch>>`

`fn declutter( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<Option<TypedModelPatch>>`

`fn codegen( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<Option<TypedModelPatch>>`

`fn fuse( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<Option<TypedModelPatch>>`

`fn cost(&self, _inputs: &[&TypedTensorInfo]) -> TractResult<TVec<(Cost, TDim)>>`

`fn nested_models(&self) -> Vec<(Cow<str>, &dyn Model)>`

`fn validation(&self) -> Validation`

`fn axes_info( &self, _model: &TypedModel, _node: &TypedNode ) -> TractResult<AxesInfo>`

`fn same_as(&self, _other: &dyn Op) -> bool`

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

`fn is_canonic(&self) -> bool`

`impl dyn Op`

`pub fn is<__T: Op>(&self) -> bool`

`pub fn downcast<T: Op>(self: Box<Self>) -> Result<Box<T>, Box<Self>>`

`pub fn downcast_ref<T: Op>(&self) -> Option<&T>`

`pub fn downcast_mut<T: Op>(&mut self) -> Option<&mut T>`

`impl AsMut<dyn Op + 'static> for dyn InferenceOp`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsMut<dyn Op + 'static> for Box<dyn InferenceOp>`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsMut<dyn Op + 'static> for dyn TypedOp`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsMut<dyn Op + 'static> for Box<dyn TypedOp>`[src]

`fn as_mut(&mut self) -> &mut dyn Op`[src]

`impl AsRef<dyn Op + 'static> for dyn InferenceOp`[src]

`fn as_ref(&self) -> &dyn Op`[src]

`impl AsRef<dyn Op + 'static> for Box<dyn InferenceOp>`[src]

`fn as_ref(&self) -> &dyn Op`[src]

`impl AsRef<dyn Op + 'static> for dyn TypedOp`[src]

`fn as_ref(&self) -> &dyn Op`[src]

`impl AsRef<dyn Op + 'static> for Box<dyn TypedOp>`[src]

`fn as_ref(&self) -> &dyn Op`[src]

`impl Op for AddDims`[src]

`fn name(&self) -> Cow<str>`[src]

`fn info(&self) -> TractResult<Vec<String>>`[src]

`fn axes_info( &self, _model: &TypedModel, node: &TypedNode ) -> TractResult<AxesInfo>`[src]

`fn is_canonic(&self) -> bool`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for Concat`[src]

`fn name(&self) -> Cow<str>`[src]

`fn declutter( &self, model: &TypedModel, node: &TypedNode ) -> TractResult<Option<TypedModelPatch>>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for ConstantLike`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for ConstantOfShape`[src]

`fn name(&self) -> Cow<str>`[src]

`fn info(&self) -> TractResult<Vec<String>>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for Crop`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for EyeLike`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for Flatten`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for Gather`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for IntoShape`[src]

`fn name(&self) -> Cow<str>`[src]

`fn info(&self) -> TractResult<Vec<String>>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for MultiBroadcastTo`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for Pad`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for PermuteAxes`[src]

`fn name(&self) -> Cow<str>`[src]

`fn is_canonic(&self) -> bool`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]

`impl Op for Reshape`[src]

`fn name(&self) -> Cow<str>`[src]

`fn as_typed(&self) -> Option<&dyn TypedOp>`[src]