validate axis argument of tf.squeeze
Hi @SuryanarayanaY Can you please check @cantonios's [comments](https://github.com/tensorflow/tensorflow/pull/62507#discussion_r1529000077)? Thank you! This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you. This PR was closed because it has been inactive for 14 days since being marked as stale. Please reopen if you'd like to work on this further.
In current implementation of `tf.squeeze`, if we pass a tensor for `axis` argument it will raise error like below, which is not inferable.

`MemoryError: std::bad_alloc`

This is due to the reason that the `Squeeze` Op expects `axis` as a `list` and there is no check for this in C++ level. Hence I am adding a validation check at Python level itself.

Attached [gist](https://colab.sandbox.google.com/gist/SuryanarayanaY/06d4e539682780b4326a4f75f654a29e/62504.ipynb#scrollTo=NeGfeG5Jnhfn) also for reference.

Might fix #62504.