Avoid evaluating SaveSpec Tensors multiple times when executing eagerly
The Saver now calls a SaveSpec callable once when saving and not at all when restoring. Previously saving evaluated the callable twice and restoring once (copying a variable's value each time). Requires a dtype be passed to a SaveSpec if its tensor is callable. PiperOrigin-RevId: 190972754
Loading
Please sign in to comment