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:

• operationThe operation that is being called.

• argsThe arguments that are passed to the operation.

• kwargsThe keyword arguments that are passed to the operation.

• outputThe output of the operation.

Returns:

Nonethe hook is executed.