Contents

MLCTrainingGraph

A training graph that you create from one or more graph objects plus additional layers you add directly to the training graph.

Declaration

class MLCTrainingGraph

Overview

The framework provides a family of graph-execution methods to execute a full training iteration, and methods to execute the forward pass, the gradient pass, and optimizer update, individually.

Use one of the execute methods to execute a full training iteration to accelerate an ML model represented as a single training graph.

Use one of the executeForward, executeGradient, or executeOptimizerUpdatemethods to accelerate an ML library that separates the forward pass, gradient pass, and optimizer update as separate phases.

Topics

Creating Training Graphs

Preparing Training Graphs

Executing Training Iterations

Executing Forward, Gradient, and Optimizer Updates

Inspecting Training Graphs

See Also

Graphs