[][src]Trait tract_hir::infer::rules::InferenceRulesOp

pub trait InferenceRulesOp {
    fn rules<'r, 'p: 'r, 's: 'r>(
        &'s self,
        solver: &mut Solver<'r>,
        inputs: &'p [TensorProxy],
        outputs: &'p [TensorProxy]
    ) -> InferenceResult;
fn as_op(&self) -> &dyn Op;
fn as_op_mut(&mut self) -> &mut dyn Op; fn to_typed(
        &self,
        source: &InferenceModel,
        node: &InferenceNode,
        target: &mut TypedModel,
        mapping: &HashMap<OutletId, OutletId>
    ) -> TractResult<TVec<OutletId>> { ... }
fn nboutputs(&self) -> TractResult<usize> { ... }
fn incorporate(
        &self,
        model: &InferenceModel,
        node: &InferenceNode
    ) -> TractResult<Option<InferenceModelPatch>> { ... } }

Required methods

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

Registers the inference rules of the operator.

fn as_op(&self) -> &dyn Op

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

Loading content...

Provided methods

fn to_typed(
    &self,
    source: &InferenceModel,
    node: &InferenceNode,
    target: &mut TypedModel,
    mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>

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

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

Loading content...

Implementors

impl InferenceRulesOp for Box<dyn Expansion>[src]

impl InferenceRulesOp for ConstantLike[src]

impl InferenceRulesOp for EyeLike[src]

impl InferenceRulesOp for Gather[src]

impl InferenceRulesOp for Pad[src]

impl InferenceRulesOp for Nary[src]

impl InferenceRulesOp for MaxPool[src]

impl InferenceRulesOp for SumPool[src]

impl InferenceRulesOp for Downsample[src]

impl InferenceRulesOp for Dummy[src]

impl InferenceRulesOp for ElementWiseOp[src]

impl InferenceRulesOp for Identity[src]

impl InferenceRulesOp for Const[src]

impl InferenceRulesOp for Iff[src]

impl InferenceRulesOp for MatMul[src]

impl InferenceRulesOp for LayerHardmax[src]

impl InferenceRulesOp for Source[src]

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

Registers the inference rules of the operator.

impl InferenceRulesOp for UnimplementedOp[src]

impl<D: DimLike + ToDim> InferenceRulesOp for Slice<D>[src]

Loading content...