Struct Tracer 

pub struct Tracer { /* private fields */ }

Records operations into an IR graph.

Implementations

impl Tracer

pub fn input(&self, name: &str, dims: &[usize], dtype: DType) -> TracedTensor

Declare a graph input with static dimensions.

pub fn input_dyn(&self, name: &str, dims: &[Dim], dtype: DType) -> TracedTensor

Declare a graph input with mixed static/dynamic dimensions.

pub fn param(&self, name: &str, dims: &[usize], dtype: DType) -> TracedTensor

Declare a parameter (weight) with static dimensions.

pub fn matmul(&self, lhs: TracedTensor, rhs: TracedTensor) -> TracedTensor

Matrix multiply.

pub fn layer_norm( &self, x: TracedTensor, gamma: TracedTensor, beta: TracedTensor, eps: f32, ) -> TracedTensor

Layer normalization.

pub fn softmax(&self, x: TracedTensor, axis: i32) -> TracedTensor

Softmax.

pub fn gather( &self, table: TracedTensor, indices: TracedTensor, axis: usize, ) -> TracedTensor

Gather (embedding lookup).