Function is_call_function

imperative/python/megengine/traced_module/expr.py:65–70

(expr, func: Iterable[Callable] = None)

def is_call_function(expr, func: Iterable[Callable] = None):
    if func and not isinstance(func, Iterable):
        func = (func,)
    return isinstance(expr, CallFunction) and (
        func is None or any(expr.func == f for f in func)
    )


def is_constant(expr):