Commit c529a131 authored by Brennan Saeta's avatar Brennan Saeta Committed by TensorFlower Gardener
Browse files

[Keras/Cloud TPU]: Place variables on the device. 

In order to improve performance, avoid transferring the weights from the host memory onto device memory every step, but instead leave weights resident on the device to be re-used on subsequent steps.

PiperOrigin-RevId: 204320023
parent e36a1d67
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