ttnn.register_post_operation_hook

ttnn.register_post_operation_hook(hook)

register_post_operation_hook is a context manager that registers a post-operation hook. The hook can be used to run custom code after the operation is executed.

Parameters:

• operation – The operation that is being called.

• args – The arguments that are passed to the operation.

• kwargs – The keyword arguments that are passed to the operation.

• output – The output of the operation.

Returns:

None – the hook is executed.