[AutoDiff] [API] Gardening for stdlib VJP definitions. (#21916)
- Do not broadcast adjoint to original value in any VJP definition (where
it is not necessary).
- Instead, assume that adjoint has same shape as original result.
- This formulation is mathematically correct. Users can manually specify
non-scalar adjoints using `pullback`.
- Gardening.
