Trait tfdeploy::analyser::rules::InferenceRulesOp

pub trait InferenceRulesOp {
    fn rules<'r, 'p: 'r, 's: 'r>(
        &'s self, 
        solver: &mut Solver<'r>, 
        inputs: &'p TensorsProxy, 
        outputs: &'p TensorsProxy
    );
}

Required Methods

fn rules<'r, 'p: 'r, 's: 'r>(
    &'s self,
    solver: &mut Solver<'r>,
    inputs: &'p TensorsProxy,
    outputs: &'p TensorsProxy
)

Registers the inference rules of the operator.

Implementors

• impl InferenceRulesOp for Const

• impl<T: Datum> InferenceRulesOp for Conv2D<T>

• impl<P: Pooler + Debug> InferenceRulesOp for Pool<P>

• impl<T: Datum> InferenceRulesOp for SpaceToBatch<T>

• impl<T: Datum> InferenceRulesOp for BatchToSpace<T>

• impl<T: Datum + Signed> InferenceRulesOp for Relu<T>

• impl<T: Datum + Float> InferenceRulesOp for Sigmoid<T>

• impl InferenceRulesOp for Softmax