megengine.utils.comp_graph_tools.replace_oprs

replace_oprs(dst, oprmap)[源代码]

替换图中的算子。

参数

  • dst (List[VarNode]) – 将会出现在新图中的目标变量。

  • oprmap (Dict[OperatorNode, OperatorNode]) – 一个指示如何替换算子的映射。

返回类型

List[VarNode]

返回

与 dst 对应且所有依赖都被替换好的新变量。

上一页

megengine.utils.comp_graph_tools.replace_vars

下一页

megengine.utils.comp_graph_tools.set_priority_to_id