Attention wrapper tests fail for float64 dtype #407
Description
Since tensorflow/tensorflow@e17b7ca the autocast dtype is causing failures. Should be a rather straight forward fix.
{{ message }}
Since tensorflow/tensorflow@e17b7ca the autocast dtype is causing failures. Should be a rather straight forward fix.