STT-tensorflow/tensorflow/python/distribute/parallel_device
Bruce Fontaine c27b834b49 Wrap save/restore logic in tf.function when in eager mode. This allows parallel saving and restoring when using multiple devices.
PiperOrigin-RevId: 317719780
Change-Id: Ifb7e34f708da4121b49fb38d8dad046d45fedc42
2020-06-22 13:23:14 -07:00
..
BUILD Add support for a device ID op in parallel_device 2020-05-12 09:34:03 -07:00
parallel_device_test.py Wrap save/restore logic in tf.function when in eager mode. This allows parallel saving and restoring when using multiple devices. 2020-06-22 13:23:14 -07:00
parallel_device.py Add support for a device ID op in parallel_device 2020-05-12 09:34:03 -07:00
pywrap_parallel_device.cc Factor out C++ types from the parallel device 2020-06-04 14:49:16 -07:00
saving.py