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]