Contents

struct

SushiAI::Train::TrainingGraph

A traced, differentiated model, ready to lower.

Declared in
include/SushiAI/train/trainer.hpp

Public attributes

Graph::Graph graph

The forward and backward IR, in one graph.

StepSpec spec

The values the loop reads and writes.

std::vector< ParameterBinding > parameters

Every parameter, paired with its gradient.

Graph::ValueId loss_scale = Graph::NO_VALUE

The 1-element leaf the backward seed is multiplied by.

ObjectiveLeaves leaves

Holds leaves declared by the objective, indexed by role.

Graph::CastReport casts

What the cast pass converted, and where.

Graph::FusionReport fusion

What the fusion pass did to graph, and what it refused.