Commit 9a20e435 authored by Jing Li's avatar Jing Li Committed by TensorFlower Gardener
Browse files

- Support bfloat16 in Keras TPU. 

- Explicitly put optimizer parameter variables on TPU.

PiperOrigin-RevId: 217544154
parent db46af2f
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