fix(get-item): Use dynamic shape rather than static (x.shape) when possible in _parse_query

[hmahmood24]

PR Description

Use dynamic shape when available rather than static x.shape e.g. when the backend is set as tensorflow in _parse_query, allows get_item to be compilable with tf.function

Related Issue

Closes #

Checklist

• Did you add a function?
• Did you add the tests?
• Did you run your tests and are your tests passing?
• Did pre-commit not fail on any check?
• Did you follow the steps we provided?

Socials

[github-actions]

PR Compliance Checks

Thank you for your Pull Request! We have run several checks on this pull request in order to make sure it's suitable for merging into this project. The results are listed in the following section.

Issue Reference

In order to be considered for merging, the pull request description must refer to a specific issue number. This is described in our contributing guide and our PR template.
This check is looking for a phrase similar to: "Fixes #XYZ" or "Resolves #XYZ" where XYZ is the issue number that this PR is meant to address.

Protected Branch

In order to be considered for merging, the pull request changes must not be implemented on the "main" branch. This is described in our Contributing Guide. We are closing this pull request and we would suggest that you implement your changes as described in our Contributing Guide and open a new pull request.

[vedpatwardhan]

lgtm! Feel free to merge, thanks @hmahmood24 😄