[URGENT] There is less inputs received while giving the exact number of inputs required 2.0

[20Saboteur20]

This is a follow up to this issue #508, which I just closed from my previous account , given that it has been just sanctioned by GitHub because I am living in a US sanctioned country , because of this I can't do anything with the previous account.
continuing with the issue , the assigned account @tilakrayal has already asked me to create a gist ; so here it is https://gist.github.com/20Saboteur20/209e8236e1655125e555d76749bc3d2b
the used files are here , https://drive.google.com/drive/folders/1728ZfGo6JUDhNODMDdC3pH6HOxqS7oiB?usp=sharing
and as a precaution here I included them as zip
Project Files.zip
.

[tilakrayal]

@gadagashwini,
I was able to reproduce the issue on tensorflow v2.8, v2.9 and nightly. Kindly find the gist of it here.

[gowthamkpr]

@20Saboteur20 This is similar to this issue. Please take a look at the solution provided here. Thanks!

[google-ml-butler]

This issue has been automatically marked as stale because it has no recent activity. It will be closed if no further activity occurs. Thank you.

[google-ml-butler]

Closing as stale. Please reopen if you'd like to work on this further.

[google-ml-butler]

Are you satisfied with the resolution of your issue?
Yes
No