Struct tract_core::ops::math::mat_mul::MatMulUnary

pub struct MatMulUnary { /* fields omitted */ }

Methods

impl MatMulUnary

pub fn new(a: Arc<Tensor>, a_trans: bool, b_trans: bool, c_trans: bool) -> Self

Constructs a new MatMulUnary.

Trait Implementations

impl StatelessOp for MatMulUnary

fn eval(&self, inputs: TVec<Arc<Tensor>>) -> TractResult<TVec<Arc<Tensor>>>

impl Op for MatMulUnary

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

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

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

fn as_pulsed(&self) -> Option<&dyn PulsedOp>

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.

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

Fuse op after codegen to deal with local optimisations.

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 same_as(&self, _other: &dyn Op) -> bool

Compare two ops.

impl TypedOp for MatMulUnary

fn output_facts(&self, inputs: &[&TypedFact]) -> TractResult<TVec<TypedFact>>

Deduce output facts from input facts.

fn axes_info(
    &self,
    model: &TypedModel,
    node: &TypedNode
) -> TractResult<AxesInfo>

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

Computes a cost hint of the operation.

fn pulsify(
    &self,
    _source: &NormalizedModel,
    node: &NormalizedNode,
    target: &mut PulsedModel,
    mapping: &HashMap<OutletId, OutletId>,
    _pulse: usize
) -> TractResult<TVec<OutletId>>

Translate an op in a normalized network (no constants) to a pulsing form, if possible.

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

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

fn as_op(&self) -> &dyn Op

Reinterpret the TypedOp as an Op.

fn as_op_mut(&mut self) -> &mut dyn Op

Reinterpret the TypedOp as an Op, mutably.

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

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

fn dispose_dummy_axis(
    &self,
    model: &TypedModel,
    node: &TypedNode,
    axis: usize
) -> TractResult<Option<Box<dyn TypedOp>>>

Transforms the op in an equivalent one, discarding one dummy axis (of dim assumed to be 1).

fn nested_model_multipliers(
    &self,
    inputs: &[&TypedFact]
) -> Vec<(Cow<str>, f32)>

Nested model multipliers, with label (for profiling).

impl PulsedOp for MatMulUnary

fn pulsed_output_facts(
    &self,
    inputs: &[&PulsedFact]
) -> TractResult<TVec<PulsedFact>>

Deduce output facts from input facts.

fn as_op(&self) -> &dyn Op

Reinterpret the PulsedOp as an Op.

fn as_op_mut(&mut self) -> &mut dyn Op

Reinterpret the PulsedOp as an Op, mutably.

fn to_typed(&self) -> Box<dyn TypedOp>

Reinterpret the PulsedOp as an TypedOp.

impl Clone for MatMulUnary

fn clone(&self) -> MatMulUnary

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MatMulUnary

fn fmt(&self, f: &mut Formatter) -> Result

Formats the value using the given formatter.