Home Original page

graph | Modular

Python package

APIs to build inference graphs for MAX Engine with Python.

Classes​

  • BufferValue: Represents a mutable semantic tensor within a Graph.
  • Graph: Represents a graph for MAX Engine.
  • KernelLibrary: Represents a library with custom ops.
  • TensorValue: Represents a value semantic tensor within a Graph.
  • Value: Represents a symbolic value within a Graph.
  • Weight: Represents a weight value in a graph.

Modules​

  • dim: APIs for graph value tensor dimensions.
  • ops: Ops you can add when staging a graph.
  • quantization: APIs to quantize graph tensors.
  • shape: APIs for graph value tensor shapes.
  • type: APIs for graph value types.
  • weights: APIs for loading weights into a graph.