Fix mixed precision compatibility issues in Bayesian Neural Networks example#2240

samudraneel05 wants to merge 3 commits intokeras-team:masterkeras-team/keras-io:masterfrom

samudraneel05:mix-precision-fixsamudraneel05/keras-io:mix-precision-fixCopy head branch name to clipboard