samba.graph

class SambaGraph(name: str)

New in version 1.18.

Base class for representing a portion of a model’s computational graph.

Parameters:

name – the name of the graph

class FwdGraph(start: Tuple[SambaTensor], stop: Tuple[SambaTensor], name: str)

Bases: SambaGraph

New in version 1.18.

Specify all or part of the forward pass of a computational graph. Multiple FwdGraphs can be combined to fully represent the forward pass of the computational graph. Use trace_multigraph to trace the graph and create a handle to the graph’s output tensors, which is required for creating a FwdGraph object.

Parameters:
  • start – the input tensors of the graph

  • stop – the output tensors of the graph

  • name – the name of the graph