megengine.utils.comp_graph_tools.get_dep_vars

get_dep_vars(var, var_type=None)

返回 var 依赖的、类型是 var_type 的 tensor.core.megbrain_graph.VarNode 列表。如果 var_type 是 None，返回所有类型。

返回类型

List[VarNode]

上一页

megengine.utils

下一页

megengine.utils.comp_graph_tools.get_owner_opr_inputs