megengine.utils.comp_graph_tools.get_oprs_seq

get_oprs_seq(outputs, prune_reshape=False, prune_immtensor=True)[源代码]

获取一个导出模型的按照拓扑序排列的算子列表。

参数

  • outputs (List[VarNode]) – 模型输出。

  • prune_reshape – 是否在推理时剪掉无用的算子。

  • prune_immtensor – whether to prune the ImmutableTensor opr.

返回类型

List[OperatorNode]

返回

按照运行顺序排列的算子列表。

上一页

megengine.utils.comp_graph_tools.graph_traversal

下一页

megengine.utils.comp_graph_tools.replace_vars