Function update

exir/pass_base.py:824–825

(key: K, args: MutableMapping[K, PyTree], schema: ArgSchema)

    kwargs = kwargs.copy()

    def update(key: K, args: MutableMapping[K, PyTree], schema: ArgSchema) -> None:
        args[key] = fn(args[key], schema)

    for i, schema in enumerate(op._schema.arguments):
        if schema.name in kwargs: