Struct tract_tensorflow::ops::logic::Merge

pub struct Merge { /* fields omitted */ }

Methods

impl Merge

pub fn new(n: usize) -> Self

Constructs a new Merge.

Trait Implementations

impl Clone for Merge

fn clone(&self) -> Merge

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Merge

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

Formats the value using the given formatter.

impl InferenceRulesOp for Merge

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 Merge

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

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 declutter(
    &self,
    _model: &Model<TypedTensorInfo>,
    _node: &Node<TypedTensorInfo>
) -> Result<Option<ModelPatch<TypedTensorInfo>>, TractError>

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 cost(
    &self,
    _inputs: &[&TypedTensorInfo]
) -> Result<SmallVec<[(Cost, TDim); 4]>, 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 Merge

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