Venkatesh_Ragavan:

17179869176.57 GiB


is quite large, so that it won’t fit on the device 
Could you post the complete shape of your inputs?
If this memory footprint is expected and is not raised by a bug, then you would have to reduce the input shape and model significantly.