megengine.traced_module.traced_module.InternalGraph.get_function_by_type

InternalGraph.get_function_by_type(func=None, recursive=True)[源代码]

Filter Exprs by the type of CallFunction.

参数

  • func (Optional[Callable]) – a built-in function, such as F.relu.

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

返回

A ExprFilterCallFunction.

上一页

megengine.traced_module.traced_module.InternalGraph.get_expr_by_id

下一页

megengine.traced_module.traced_module.InternalGraph.get_method_by_type