Struct tract_core::ops::cnn::ConvUnary

pub struct ConvUnary {
    pub data_format: DataFormat,
    pub kernel_fmt: KernelFormat,
    pub padding: PaddingSpec,
    pub dilations: TVec<usize>,
    pub strides: TVec<usize>,
    pub kernel: Tensor,
    pub bias: Option<Tensor>,
    pub full_input_shape: TVec<TDim>,
    pub full_output_shape: TVec<TDim>,
    pub group: usize,
}

Fields

data_format: DataFormat

kernel_fmt: KernelFormat

padding: PaddingSpec

dilations: TVec<usize>

strides: TVec<usize>

kernel: Tensor

bias: Option<Tensor>

full_input_shape: TVec<TDim>

full_output_shape: TVec<TDim>

group: usize

Methods

impl ConvUnary

pub fn new(
    conv: &Conv,
    full_input_shape: &[TDim],
    full_output_shape: &[TDim],
    kernel: Tensor,
    bias: Option<Tensor>,
    group: usize
) -> TractResult<ConvUnary>

pub fn to_direct(&self, input_full_shape: &[usize]) -> TractResult<Direct>

pub fn to_boxed_im2col_pair<T>(
    &self,
    input_full_shape: &[usize]
) -> TractResult<(Box<dyn Op>, TVec<usize>, Box<dyn Op>)> where
    T: Datum + Clone + LinalgScalar + AddAssign<T> + FloatLike, 

pub fn rm_dummy_axis(&self, axis: usize) -> TractResult<Option<ConvUnary>>

pub fn to_depth_wise<T>(&self, shape: &[usize]) -> TractResult<Box<dyn Op>> where
    T: Datum + Clone + LinalgScalar + AddAssign<T> + PartialEq + Sum, 

Trait Implementations

impl InferenceRulesOp for ConvUnary

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 ConvUnary

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

impl Op for ConvUnary

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

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

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

fn codegen(
    &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 infer(
    &self,
    inputs: TVec<&TensorFact>,
    outputs: TVec<&TensorFact>
) -> TractResult<(TVec<TensorFact>, TVec<TensorFact>)>

Infers properties about the input and output tensors.

fn rounding_errors(&self) -> bool

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

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

impl Clone for ConvUnary

fn clone(&self) -> ConvUnary

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ConvUnary

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

Formats the value using the given formatter.