megengine.utils

Computing Graph Tools

get_dep_vars	返回 var 依赖的、类型是 var_type 的 tensor.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.

Hook

HookHandler

Network

Network

Convert

as_varnode	将一个 VarNode 兼容对象转换成 VarNode.
as_oprnode	convert a OpNode compatible object to OpNode; it works like as_varnode.i

NodeFilter

NodeFilter

Filter on node iterator.

Others

NodeFilterType	参考 NodeFilter.type.
NodeFilterNotType	参考 NodeFilter.not_type.
NodeFilterCheckType	see NodeFilter.check_type
NodeFilterHasInput	参考 NodeFilter.has_input
NodeFilterName	参考 NodeFilter.name。

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模式下对图执行进行性能分析。

上一页

megengine.hub.import_module

下一页

megengine.utils.comp_graph_tools.get_dep_vars