Struct tract_core::ops::cnn::Conv

pub struct Conv {
    pub data_format: DataFormat,
    pub kernel_fmt: KernelFormat,
    pub dilations: Option<TVec<usize>>,
    pub kernel_shape: Option<TVec<usize>>,
    pub padding: PaddingSpec,
    pub strides: Option<TVec<usize>>,
    pub group: usize,
}

Fields

data_format: DataFormat

kernel_fmt: KernelFormat

dilations: Option<TVec<usize>>

kernel_shape: Option<TVec<usize>>

padding: PaddingSpec

strides: Option<TVec<usize>>

group: usize

Methods

impl Conv

pub fn new(
    data_format: DataFormat,
    kernel_fmt: KernelFormat,
    dilations: Option<TVec<usize>>,
    kernel_shape: Option<TVec<usize>>,
    padding: PaddingSpec,
    strides: Option<TVec<usize>>,
    group: usize
) -> Self

Constructs a new Conv.

impl Conv

pub fn output_shape<D: DimLike>(
    &self,
    ishape: &[D],
    kshape: &[usize]
) -> TVec<D>

pub fn to_unary(
    &self,
    inputs: &[impl Borrow<TypedTensorInfo>]
) -> TractResult<Option<ConvUnary>>

pub fn add_bias_t<T: FloatLike + AddAssign>(
    &self,
    conv_result: &mut Tensor,
    bias: &Tensor
) -> TractResult<()>

Trait Implementations

impl InferenceRulesOp for Conv

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.

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>

impl StatelessOp for Conv

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

impl Op for Conv

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

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

Computes a cost hint of the operation.

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

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

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

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 codegen(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

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

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 axes_info(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<AxesInfo>

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

Compare two ops.

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

impl TypedOp for Conv

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 output_facts(
    &self,
    inputs: &[&TypedTensorInfo]
) -> TractResult<TVec<TypedTensorInfo>>

Deduce output facts from input facts.

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 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 nested_model_multipliers(
    &self,
    inputs: &[&TypedTensorInfo]
) -> Vec<(Cow<str>, f32)>

Nested model multipliers, with label (for profiling).

impl Default for Conv

fn default() -> Conv

Returns the "default value" for a type.

impl Clone for Conv

fn clone(&self) -> Conv

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Conv

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

Formats the value using the given formatter.