Commit b3b8dff0 authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Checkpointing for OptimizerV2 

Copies and pastes the existing Optimizer checkpointing code, and stops adding unconditional dependencies on slot variables (which were based on ops.uid() and so not reproducible across program runs).

PiperOrigin-RevId: 225248820
parent 00151d81
Expand all 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