package

checkpoint

APIs to save and load checkpoints for MAX graphs.

Modules

• ​save_load: Defines functions to save/load tensors from a checkpoint file.
• ​tensor_dict: Structs used for maintaining a collection of tensors.