It seems like there is some compatibility conflicts with your libraries. I tried to replicate the code and its working fine in the latest versions of tensorflow and keras. So, Please try to upgrade the tensorflow to the latest version. Kindly refer to this gist and this Tested build configurations to use compatible versions.