STT-tensorflow/tensorflow/python/training/tracking
Ken Franko b8694e39d8 Correctly set the experimental_io_device when restoring variable from a checkpoint.
PiperOrigin-RevId: 320222381
Change-Id: I30187c7777ab8056e48004ef5e4ae747edc32227
2020-07-13 15:50:48 -07:00
..
base_test.py Ignore get_configs that aren't JSON-serializable when saving checkpoints 2019-02-22 13:32:05 -08:00
base.py Correctly set the experimental_io_device when restoring variable from a checkpoint. 2020-07-13 15:50:48 -07:00
benchmarks_test.py xrange() was removed from Python on 1/1/2020 2020-06-08 21:45:46 +02:00
BUILD Move tracking_util_xla_test to keras/tests 2020-06-15 13:24:58 -07:00
data_structures_test.py Fork keras related tracking test to keras/tests 2020-06-15 09:38:41 -07:00
data_structures.py fix kokoro build :deprecated message. 2020-06-15 14:23:11 -07:00
graph_view.py Add SaveableObjects to SavedModel. 2020-07-01 14:28:41 -07:00
layer_utils.py Optimize prediction overhead by adding a sub-layer aware attribute cache, as well as various other minor optimizations. (Such as caching a layer's full argspec.) 2019-09-11 16:46:47 -07:00
python_state_test.py Switch _NumpyState unit tests to public APIs 2019-02-26 17:26:33 -08:00
python_state.py Adds tf.train.experimental.PythonState 2019-02-25 12:47:48 -08:00
tracking_test.py Fork keras related tracking test to keras/tests 2020-06-15 09:38:41 -07:00
tracking.py Make "map_resources" overridable by subclass of Trackable. 2020-06-18 15:59:43 -07:00
util_test.py Fork keras related tracking test to keras/tests 2020-06-15 09:38:41 -07:00
util_with_v1_optimizers_test.py Fork tracking/util_with_v1_optimizers_test to keras/tests. 2020-06-15 10:49:59 -07:00
util.py Add an option to choose the I/O Device for saving and loading models. 2020-04-22 11:27:15 -07:00