Commit f22e833a authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Initial support for mixed precision policies. 



A tf.keras.mixed_precision.experimental.Policy determines the dtype of layer computations and variables. In this initial implementation, policies only determine layer variable dtypes, and support for determining layer computation dtypes will come later.

Co-authored-by: default avatarJames Qin <jamesqin@google.com>
PiperOrigin-RevId: 236010048
parent 97e36fc6
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment