pub enum LloOp {
Elementwise {
kind: ElementwiseKind,
inputs: Vec<usize>,
output_shape: Vec<usize>,
strategy: ElementwiseStrategy,
},
Reduction {
axis: Option<usize>,
inputs: Vec<usize>,
output_shape: Vec<usize>,
},
MatMul {
a: usize,
b: usize,
output_shape: Vec<usize>,
},
Shape {
kind: ShapeKind,
inputs: Vec<usize>,
output_shape: Vec<usize>,
},
Stats {
kind: StatsKind,
inputs: Vec<usize>,
output_shape: Vec<usize>,
},
Random {
kind: RandomKind,
inputs: Vec<usize>,
output_shape: Vec<usize>,
},
Model {
kind: ModelKind,
inputs: Vec<usize>,
output_shape: Vec<usize>,
},
Training {
kind: TrainingKind,
inputs: Vec<usize>,
output_shape: Vec<usize>,
},
}Expand description
Enumeration of LLO operations. Each variant holds a backend-ready shape and input references
Variants§
Elementwise
Reduction
MatMul
Shape
Shape-related operations (reshape, transpose, concat, etc.)
Stats
Higher-level stats ops (softmax, norm, cross-entropy, …)
Random
Random number generation ops (rand, randn, randint, seed)
Model
Model-related operations (save, load, training)
Training
Training operations (forward, backward, update)
Trait Implementations§
Source§impl<'de> Deserialize<'de> for LloOp
impl<'de> Deserialize<'de> for LloOp
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for LloOp
impl RefUnwindSafe for LloOp
impl Send for LloOp
impl Sync for LloOp
impl Unpin for LloOp
impl UnwindSafe for LloOp
Blanket Implementations§
§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§unsafe fn clone_to_uninit(&self, dest: *mut u8)
unsafe fn clone_to_uninit(&self, dest: *mut u8)
🔬This is a nightly-only experimental API. (
clone_to_uninit)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