HOOKS# Functions# models.tak_utils.hooks.hook_backward(module, _, grad_output)[source]# models.tak_utils.hooks.hook_backward_cls_token(module, _, grad_output)[source]# models.tak_utils.hooks.hook_backward_layer_norm(module, _, grad_output)[source]# models.tak_utils.hooks.hook_backward_nosequence(module, _, grad_output)[source]# models.tak_utils.hooks.hook_forward_batch(module, inputs, _)[source]# models.tak_utils.hooks.hook_forward_batch_nosequence(module, inputs, _)[source]# models.tak_utils.hooks.hook_forward_layer_norm_batch(module, inputs, _)[source]# models.tak_utils.hooks.hook_forward_store_inputs(module, inputs, _)[source]# models.tak_utils.hooks.hook_backward(module, _, grad_output)[source]# models.tak_utils.hooks.hook_backward_cls_token(module, _, grad_output)[source]# models.tak_utils.hooks.hook_backward_layer_norm(module, _, grad_output)[source]# models.tak_utils.hooks.hook_backward_nosequence(module, _, grad_output)[source]# models.tak_utils.hooks.hook_forward_batch(module, inputs, _)[source]# models.tak_utils.hooks.hook_forward_batch_nosequence(module, inputs, _)[source]# models.tak_utils.hooks.hook_forward_layer_norm_batch(module, inputs, _)[source]# models.tak_utils.hooks.hook_forward_store_inputs(module, inputs, _)[source]#