Extract

The module of TVM auto_scheduler workload extraction from DNN models.

lorien.dialect.tvm_dial.auto_scheduler_dial.extract.extract_from_models(configs: argparse.Namespace)

Extract op workloads from a given model.

Parameters

configs (argparse.Namespace) -- The system configure of generate.extract-from-model.

Returns

workloads -- A list of collected workloads.

Return type

List[Workload]