Module exec 

Graph executor — runs .sw programs on the tensor runtime.

Re-exports

pub use jit::compile_graph;

pub use jit::load_jit;

pub use jit::CompileStats;

pub use jit::CompiledGraph;

pub use jit::Instruction;

pub use jit::JitExecutor;

pub use jit::JitResult;

pub use jit::MemoryPlan;

Modules

jit

Structs

EpochLog

Log for a single training epoch.

ExecResult

The result of executing a graph.

Executor

Executes IrProgram graphs on the Shrew tensor runtime.

RuntimeConfig

Runtime configuration for resolving symbolic dimensions and execution mode.

TrainResult

Summary of a full training run.

Trainer

High-level training loop runner.

Functions

load_program

Parse, lower, validate, optimize, and prepare an executor from .sw source.

load_trainer

Parse, lower, validate, optimize, and prepare a trainer from .sw source.