pub enum DomainInstruction {
Neural(NeuralNode),
MatMul {
a_shape: Vec<usize>,
b_shape: Vec<usize>,
transpose_a: bool,
transpose_b: bool,
},
Conv2D {
stride: [usize; 2],
padding: [usize; 2],
dilation: [usize; 2],
groups: usize,
},
ElementWise(GaiaType, String),
GetThreadId(usize),
GetGroupSize(usize),
Barrier,
}Expand description
Tier 2: 领域特定指令 (神经网络/张量/并行计算)
Variants§
Neural(NeuralNode)
神经网络算子
MatMul
矩阵乘法 (A, B, C)
Conv2D
卷积运算
ElementWise(GaiaType, String)
逐元素运算 (类型, 算子名)
GetThreadId(usize)
获取线程 ID (维度)
GetGroupSize(usize)
获取工作组大小 (维度)
Barrier
屏障同步
Trait Implementations§
Source§impl Clone for DomainInstruction
impl Clone for DomainInstruction
Source§fn clone(&self) -> DomainInstruction
fn clone(&self) -> DomainInstruction
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for DomainInstruction
impl Debug for DomainInstruction
Source§impl<'de> Deserialize<'de> for DomainInstruction
impl<'de> Deserialize<'de> for DomainInstruction
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
Source§impl PartialEq for DomainInstruction
impl PartialEq for DomainInstruction
Source§impl Serialize for DomainInstruction
impl Serialize for DomainInstruction
impl StructuralPartialEq for DomainInstruction
Auto Trait Implementations§
impl Freeze for DomainInstruction
impl RefUnwindSafe for DomainInstruction
impl Send for DomainInstruction
impl Sync for DomainInstruction
impl Unpin for DomainInstruction
impl UnsafeUnpin for DomainInstruction
impl UnwindSafe for DomainInstruction
Blanket Implementations§
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