afe.core.parse_networks

Functions

import_model_from_model_configs(...)

Importing a model written in any of the supported libraries from a file,

Module Contents

afe.core.parse_networks.import_model_from_model_configs(config: afe.core.configs.ModelConfigs) → Tuple[afe._tvm._defines.TVMIRModule, List[str] | None]

Importing a model written in any of the supported libraries from a file, and returning it as TVM IR module.

Parameters:

config – ModelConfigs of the model being loaded. Contains information like path, framework and other.

Returns:

Tuple of the model converted to TVM IR module and the model’s output names.