swift

[TF] Refactor gradients computation with expandingShape

#24249

Merged