Extract From Model¶
The module of AutoTVM workload extraction from DNN models.
- lorien.dialect.tvm_dial.autotvm_dial.extract_from_model.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]