Struct MatmulEpilogue 

pub struct MatmulEpilogue {
    pub alpha: f64,
    pub beta: f64,
    pub row_scale: Option<GpuTensorHandle>,
    pub col_scale: Option<GpuTensorHandle>,
    pub row_op: ScaleOp,
    pub col_op: ScaleOp,
    pub clamp_min: Option<f64>,
    pub clamp_max: Option<f64>,
    pub pow_exponent: Option<f64>,
    pub diag_output: Option<GpuTensorHandle>,
}

Fields

alpha: f64

Scalar multiply applied to each output element.

beta: f64

Scalar add applied to each output element after scaling.

row_scale: Option<GpuTensorHandle>

Optional per-row scale (length m). When present, output[row, col] *= row_scale[row].

col_scale: Option<GpuTensorHandle>

Optional per-column scale (length n). When present, output[row, col] *= col_scale[col].

row_op: ScaleOp

Row scale operation (multiply or divide). Ignored when row_scale is None.

col_op: ScaleOp

Column scale operation (multiply or divide). Ignored when col_scale is None.

clamp_min: Option<f64>

Optional lower clamp bound applied after scale/bias.

clamp_max: Option<f64>

Optional upper clamp bound applied after scale/bias.

pow_exponent: Option<f64>

Optional power exponent applied after clamp (final operation in the epilogue).

diag_output: Option<GpuTensorHandle>

Optional output buffer for the diagonal of the result (length min(m, n)).

Implementations

impl MatmulEpilogue

pub fn noop() -> Self

pub fn is_noop(&self) -> bool

Trait Implementations

impl Clone for MatmulEpilogue

fn clone(&self) -> MatmulEpilogue

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MatmulEpilogue

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for MatmulEpilogue

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for MatmulEpilogue

fn eq(&self, other: &MatmulEpilogue) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for MatmulEpilogue

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for MatmulEpilogue