Struct TensorNode 

pub struct TensorNode {
    pub op: TensorOp,
    pub data_ptr: u64,
    pub shape: [u32; 4],
    pub inputs: Vec<usize>,
    pub params: OpParams,
}

A node in the compute graph.

Each node represents a tensor operation with device-memory pointers to its data and references to input nodes (by index).

Fields

op: TensorOp

Operation to perform

data_ptr: u64

Device pointer to output data

shape: [u32; 4]

Output dimensions [rows, cols, batch, unused]

inputs: Vec<usize>

Indices of input nodes in the graph (max 3: src0, src1, src2)

params: OpParams

Operation-specific parameters (e.g., epsilon for RmsNorm, position for RoPE)

Trait Implementations

impl Clone for TensorNode

fn clone(&self) -> TensorNode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorNode

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

Formats the value using the given formatter.