Struct tract_core::ops::cnn::conv::Direct

pub struct Direct { /* fields omitted */ }

Methods

impl Direct

pub fn new(
    conv: Box<dyn Conv<f32>>,
    input_shape: DataShape,
    output_shape: DataShape,
    packed_filters: Tensor
) -> Self

Constructs a new Direct.

Trait Implementations

impl InferenceRulesOp for Direct

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

Registers the inference rules of the operator.

impl StatelessOp for Direct

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

impl Op for Direct

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

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

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

fn rounding_errors(&self) -> bool

fn infer(
    &self,
    inputs: TVec<&TensorFact>,
    outputs: TVec<&TensorFact>
) -> TractResult<(TVec<TensorFact>, TVec<TensorFact>)>

Infers properties about the input and output tensors.

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

fn pulsify(
    &self,
    _source: &NormalizedModel,
    _node: &NormalizedNode,
    _target: &mut PulsedModel,
    _mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>

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

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

impl Clone for Direct

fn clone(&self) -> Direct

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Direct

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

Formats the value using the given formatter.