set_value_magic_handler

onnx_ir.set_value_magic_handler(handler)

Set the magic handler for Value arithmetic methods.

Framework authors can implement custom context managers that set the magic handler to enable arithmetic operations on Values.

Example:

class MyOpHandler:
    def Add(self, lhs, rhs):
        # Implement addition logic here
        pass
    ...

@contextlib.contextmanager
def graph_context(graph):
    old_handler = onnx_ir.set_value_magic_handler(MyOpHandler(graph))
    try:
        yield
    finally:
        onnx_ir.set_value_magic_handler(old_handler)

Parameters:

handler (_OpHandlerProtocol | None) – The magic handler to set.

Returns:

The previous magic handler.

Return type:

_OpHandlerProtocol | None