Struct tract_tensorflow::ops::nn::dw_conv2d::DepthwiseConv2d

pub struct DepthwiseConv2d { /* fields omitted */ }

Methods

impl DepthwiseConv2d

pub fn new(
    data_format: DataFormat,
    padding: PaddingSpec,
    strides: TVec<usize>,
    dilations: TVec<usize>
) -> Self

Constructs a new DepthwiseConv2d.

Trait Implementations

impl Clone for DepthwiseConv2d

fn clone(&self) -> DepthwiseConv2d

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for DepthwiseConv2d

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

Formats the value using the given formatter.

impl InferenceRulesOp for DepthwiseConv2d

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 Op for DepthwiseConv2d

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 infer(
    &self,
    inputs: SmallVec<[&TensorFact; 4]>,
    outputs: SmallVec<[&TensorFact; 4]>
) -> Result<(SmallVec<[TensorFact; 4]>, SmallVec<[TensorFact; 4]>), TractError>

Infers properties about the input and output tensors.

fn pulsify(
    &self,
    _source: &Model<NormalizedTensorInfo>,
    _node: &Node<NormalizedTensorInfo>,
    _target: &mut Model<PulsedTensorFact>,
    _mapping: &HashMap<OutletId, OutletId, RandomState>
) -> Result<SmallVec<[OutletId; 4]>, TractError>

fn codegen(
    &self,
    _model: &Model<TypedTensorInfo>,
    _node: &Node<TypedTensorInfo>
) -> Result<Option<ModelPatch<TypedTensorInfo>>, TractError>

fn rounding_errors(&self) -> bool

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

fn info(&self) -> Result<Option<String>, TractError>

impl StatelessOp for DepthwiseConv2d

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