Enum DomainInstruction 

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,
}

Tier 2: Domain-specific instructions (Neural network/Tensor/Parallel computing)

Variants

Neural(NeuralNode)

Neural network operator node.

MatMul

Matrix multiplication operation.

Fields

a_shape: Vec<usize>

The shape of matrix A.

b_shape: Vec<usize>

The shape of matrix B.

transpose_a: bool

Whether to transpose matrix A before multiplication.

transpose_b: bool

Whether to transpose matrix B before multiplication.

Conv2D

2D Convolution operation.

Fields

stride: [usize; 2]

The stride values [height, width].

padding: [usize; 2]

The padding values [height, width].

dilation: [usize; 2]

The dilation values [height, width].

groups: usize

The number of groups for grouped convolution.

ElementWise(GaiaType, String)

Element-wise operation on tensors.

GetThreadId(usize)

Get the thread ID in the specified dimension.

GetGroupSize(usize)

Get the group size in the specified dimension.

Barrier

Barrier synchronization for all threads in a group.

Trait Implementations

impl Clone for DomainInstruction

fn clone(&self) -> DomainInstruction

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DomainInstruction

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for DomainInstruction

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for DomainInstruction

fn eq(&self, other: &DomainInstruction) -> 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 DomainInstruction

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 DomainInstruction