Unable to convert some input_tensors to shard

[punithsekar]

Describe the bug Unable to convert the following [361, 49, 96],[100, 49, 192], [25, 49, 384], [9, 49, 768] input shapes to shard tensors. since the 1 Dimension is odd.

To Reproduce Steps to reproduce the behavior:

1. Checkout to branch punith/swin_s_MM_input_shard Expected behavior
2. Run command,pytest models/experimental/functional_swin_s/tt/unit_test_MM_input.py

Please complete the following environment information:

• OS: Ubuntu 20.04
• Device: WH-n150

[punithsekar]

fyi @saichandax