megengine.utils

Computing Graph Tools

get_dep_vars

返回 var 依赖的、类型是 var_typetensor.core.megbrain_graph.VarNode 列表。

get_owner_opr_inputs

获取拥有这个变量的算子的所有输入。

get_owner_opr_type

获得拥有这个变量的算子类型。

get_opr_type

获取一个算子的类型。

graph_traversal

一个用于遍历计算图并返回足够有用的信息的辅助函数。

get_oprs_seq

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

replace_vars

替换图中的变量。

replace_oprs

替换图中的算子。

set_priority_to_id

For all oprs in the subgraph constructed by dest_vars, sets its priority to id if its original priority is zero.

GraphInference

Loads a serialized computing graph as a GraphInference object which can be used to execute the computing graph.

Network

Network

Convert

as_varnode

将一个 VarNode 兼容对象转换成 VarNode.

as_oprnode

将一个 OpNode 兼容对象转换成 OpNode; 效果如同 as_varnode.

NodeFilter

NodeFilter

Filter on node iterator.

Network Node

network_node

Module Stats

module_stats

Calculate and print model’s statistics by adding hook and record Module’s inputs outputs size.

Profiler

Profiler

imperative模式下对图执行进行性能分析。