Debugging : Keras+tensorflow code to create sequential model crashing with message “Your session crashed for an unknown reason” on Google colab

Question

While running the following code on Google colab with \'GPU\' runtime (as one of my custom layers performs tensorflow.fft using GPU), my session is crashing -