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

…