megengine.utils.comp_graph_tools.replace_vars

replace_vars(dst, varmap)[源代码]

替换图中的变量。

参数

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

  • varmap (Dict[VarNode, VarNode]) – 一个指示如何替换变量的映射。

返回类型

List[VarNode]

返回

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

上一页

megengine.utils.comp_graph_tools.get_oprs_seq

下一页

megengine.utils.comp_graph_tools.replace_oprs