google / edward2

A simple probabilistic programming language.
Apache License 2.0
679 stars 75 forks source link

Fixes an edge case in which the input dtype is not equal to the compute dtype. #548

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Fixes an edge case in which the input dtype is not equal to the compute dtype.

The underlying tf.keras.layers.Dense layer has the same conditional statement.