ViT and Swin symbolic tracing with torch.fx
# What does this PR do?

This PR adds support for ViT and Swin symbolic tracing with torch.fx.

Fixes #16320




_The documentation is not available anymore as the PR was closed or merged._