Trait tract_pulse::internal::TypedOp

pub trait TypedOp: Op + Debug + DynClone + Send + Sync + 'static + Downcast + EvalOp + DynHash {

Show 15 methods    fn as_op(&self) -> &(dyn Op + 'static);
    fn as_op_mut(&mut self) -> &mut (dyn Op + 'static);
    fn output_facts(
        &self,
        inputs: &[&TypedFact]
    ) -> Result<SmallVec<[TypedFact; 4]>, Error>;

    fn invariants(
        &self,
        inputs: &[&TypedFact],
        outputs: &[&TypedFact]
    ) -> Result<Invariants, Error> { ... }
    fn fuse(
        &self,
        _model: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        _node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>
    ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static, Global>>>, Error> { ... }
    fn declutter_with_session(
        &self,
        session: &mut OptimizerSession<'_>,
        model: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>
    ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static, Global>>>, Error> { ... }
    fn declutter(
        &self,
        model: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>
    ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static, Global>>>, Error> { ... }
    fn cost(
        &self,
        _inputs: &[&TypedFact]
    ) -> Result<SmallVec<[(Cost, TDim); 4]>, Error> { ... }
    fn suggested_axis_changes(
        &self
    ) -> Result<SmallVec<[(InOut, AxisOp); 4]>, Error> { ... }
    fn change_axes(
        &self,
        model: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        io: InOut,
        change: &AxisOp
    ) -> Result<Option<AxisChangeConsequence>, Error> { ... }
    fn slice(
        &self,
        patch: &mut ModelPatch<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        prefix: &str,
        inputs: &[OutletId],
        output_axis: usize,
        start: usize,
        end: usize
    ) -> Result<Option<SmallVec<[OutletId; 4]>>, Error> { ... }
    fn quantize(
        &self,
        model: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        dt: DatumType,
        scale: f32,
        zero_point: i32
    ) -> Result<Option<Box<dyn TypedOp + 'static, Global>>, Error> { ... }
    fn concretize_dims(
        &self,
        source: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        target: &mut Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        mapping: &HashMap<OutletId, OutletId, RandomState>,
        values: &SymbolValues
    ) -> Result<SmallVec<[OutletId; 4]>, Error> { ... }
    fn codegen(
        &self,
        model: &Graph<TypedFact, Box<dyn TypedOp + 'static, Global>>,
        node: &Node<TypedFact, Box<dyn TypedOp + 'static, Global>>
    ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp + 'static, Global>>>, Error> { ... }
    fn nested_model_multipliers(
        &self,
        inputs: &[&TypedFact]
    ) -> Vec<(Cow<'_, str>, f64), Global> ⓘ { ... }

}

Required Methods§

Reinterpret the TypedOp as an Op.

Reinterpret the TypedOp as an Op, mutably.

Deduce output facts from input facts.

Provided Methods§

Fuse op after codegen to deal with local optimisations.

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

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

Computes a cost hint of the operation.

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

Transforms the op in an equivalent one, operating on dt (i8 or u8).

Returns None if the op can not be translated.

Transform the op into by providing a value to one or more symbols.

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.

Nested model multipliers, with label (for profiling).

Implementations§

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

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.

Returns an Rc-ed object from an Rc-ed trait object if the underlying object is of type __T. Returns the original Rc-ed trait if it isn’t.

Returns a reference to the object within the trait object if it is of type __T, or None if it isn’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§

Converts this type into a mutable reference of the (usually inferred) input type.
Converts this type into a mutable reference of the (usually inferred) input type.
Converts this type into a shared reference of the (usually inferred) input type.
Converts this type into a shared reference of the (usually inferred) input type.
Returns a copy of the value. Read more
Performs copy-assignment from source. Read more
Returns a copy of the value. Read more
Performs copy-assignment from source. Read more
Returns a copy of the value. Read more
Performs copy-assignment from source. Read more
Returns a copy of the value. Read more
Performs copy-assignment from source. Read more
Formats the value using the given formatter. Read more
Converts to this type from the input type.
Converts to this type from the input type.
Converts to this type from the input type.
Feeds this value into the given Hasher. Read more
Feeds a slice of this type into the given Hasher. Read more

Implementations on Foreign Types§

Implementors§