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]