Python package

graph

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.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit