Struct tract_core::ops::nn::ConvUnary

pub struct ConvUnary {
    pub data_fmt: 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_fmt: 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 rm_dummy_axis(&self, axis: usize) -> TractResult<Option<ConvUnary>>

Trait Implementations

impl InferenceRulesOp for ConvUnary

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

Registers the inference rules of the operator.

impl StatelessOp for ConvUnary

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

impl Op for ConvUnary

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

fn reduce(     &self,     inputs: TVec<&TensorFact>,     _outputs: TVec<&TensorFact>,     phase: ReductionPhase ) -> TractResult<Option<ReducedOpRewire>>

fn pulsify(     &self,     inputs: TVec<&PulsedTensorFact> ) -> TractResult<Vec<PulsifiedOp>>

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

Infers properties about the input and output tensors.

fn const_value(&self) -> Option<SharedTensor>

fn rounding_errors(&self) -> bool

fn noutputs(&self) -> usize

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.