pub enum FusedOp {
ScaleBias {
scale: f64,
bias: f64,
},
ScaleReluBias {
scale: f64,
bias: f64,
},
ClampNormalize {
min: f64,
max: f64,
},
Identity,
Passthrough(TensorOp),
}Expand description
A fused (compound) operation produced by TensorOpFusion::fuse.
Variants§
ScaleBias
Scale followed immediately by Bias: y = x * scale + bias.
ScaleReluBias
Scale, then ReLU, then Bias: y = max(0, x * scale) + bias.
ClampNormalize
Clamp followed by L2-normalisation.
Identity
No-op — used when the input sequence was empty.
Passthrough(TensorOp)
A single operation that could not be fused with its neighbours.
Trait Implementations§
impl StructuralPartialEq for FusedOp
Auto Trait Implementations§
impl Freeze for FusedOp
impl RefUnwindSafe for FusedOp
impl Send for FusedOp
impl Sync for FusedOp
impl Unpin for FusedOp
impl UnsafeUnpin for FusedOp
impl UnwindSafe for FusedOp
Blanket Implementations§
impl<T> Allocation for T
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more