docs: update optimizer_zero_grad order, and the backward pass. (#21144)

Author: GdoongMathew

docs/source-pytorch/common/hooks.rst

@@ -143,23 +143,24 @@ with the source of each hook indicated:
  │ │ │ ├── [LightningModule]
  │ │ │ └── [Strategy]
  │ │ │
- │ │ ├── on_before_zero_grad()
- │ │ │ ├── [Callbacks]
- │ │ │ └── [LightningModule]
- │ │ │
  │ │ ├── [Forward Pass - training_step()]
  │ │ │ └── [Strategy only]
  │ │ │
- │ │ ├── on_before_backward()
+ │ │ ├── on_before_zero_grad()
  │ │ │ ├── [Callbacks]
  │ │ │ └── [LightningModule]
  │ │ │
- │ │ ├── [Backward Pass]
- │ │ │ └── [Strategy only]
+ │ │ ├── optimizer_zero_grad()
+ │ │ │ └── [LightningModule only - optimizer_zero_grad()]
  │ │ │
- │ │ ├── on_after_backward()
- │ │ │ ├── [Callbacks]
- │ │ │ └── [LightningModule]
+ │ │ ├── [Backward Pass - Strategy.backward()]
+ │ │ │ ├── on_before_backward()
+ │ │ │ │ ├── [Callbacks]
+ │ │ │ │ └── [LightningModule]
+ │ │ │ ├── LightningModule.backward()
+ │ │ │ └── on_after_backward()
+ │ │ │ ├── [Callbacks]
+ │ │ │ └── [LightningModule]
  │ │ │
  │ │ ├── on_before_optimizer_step()
  │ │ │ ├── [Callbacks]