megengine.traced_module.traced_module.InternalGraph.get_module_by_type

InternalGraph.get_module_by_type(module_cls, recursive=True)[源代码]

Filter Nodes by the module_type of ModuleNode.

参数

  • module_cls (Module) – a subclass of Module.

  • recursive – whether to get the Nodes in the subgraph. Default: True

返回

A NodeFilterType.

上一页

megengine.traced_module.traced_module.InternalGraph.get_method_by_type

下一页

megengine.traced_module.traced_module.InternalGraph.get_node_by_id