Struct tract_core::ops::math::gemm::GemmUnaryB

pub struct GemmUnaryB { /* fields omitted */ }

Methods

impl GemmUnaryB

pub fn new(
    alpha: f32,
    beta: f32,
    trans_a: bool,
    trans_b: bool,
    a: Arc<Tensor>,
    c: Arc<Tensor>
) -> Self

Constructs a new GemmUnaryB.

Trait Implementations

impl InferenceRulesOp for GemmUnaryB

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 GemmUnaryB

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

impl Op for GemmUnaryB

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

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

Infers properties about the input and output tensors.

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

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

fn rounding_errors(&self) -> bool

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

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

impl Clone for GemmUnaryB

fn clone(&self) -> GemmUnaryB

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for GemmUnaryB

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

Formats the value using the given formatter.