commit d83a344d7695dac59262716c54962c3f6c19c09b
author Guangda Lai <laigd@google.com> Mon Mar 30 13:58:20 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Mar 30 14:01:01 2020 -0700
tree 3b868e540ac595a4d2f538bbbe268db9e70cb4fc
parent f9a22e49b4a6dacf89f9cc83e07ee9f1a3ca60ff

Mark bfloat16 as trainable.

PiperOrigin-RevId: 303816514
Change-Id: Iebaed635ea7adaf8ded604aba4d81907bce86395

tensorflow/python/eager/backprop_util.py

diff --git a/tensorflow/python/eager/backprop_util.py b/tensorflow/python/eager/backprop_util.py
index ae026c0..117b05e 100644
--- a/tensorflow/python/eager/backprop_util.py
+++ b/tensorflow/python/eager/backprop_util.py
@@ -30,4 +30,4 @@
   dtype = dtypes.as_dtype(dtype)
   return dtype.base_dtype in (dtypes.float16, dtypes.float32, dtypes.float64,
                               dtypes.complex64, dtypes.complex128,
-                              dtypes.resource, dtypes.variant)
+                              dtypes.resource, dtypes.variant, dtypes.bfloat16)