diff options
| author |  Derek Murray <mrry@google.com> | 2018-07-19 11:27:39 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-19 11:35:55 -0700 |
| commit | 109ae67a7e99e3dcb4d93cc22df5b3912f4558c9 (patch) | |
| tree | 7e8f56cba34b6d6bfb6be34610ab35a968dd0db0 /tensorflow/c | |
| parent | de6be2cbb1d2a179fc8abe984c21f968df627eaf (diff) | |
Expose each device's incarnation via `Session.list_devices()`.
PiperOrigin-RevId: 205273020
Diffstat (limited to 'tensorflow/c')
| -rw-r--r-- | tensorflow/c/c_api.cc | 1 | ||||
| -rw-r--r-- | tensorflow/c/c_api.h | 7 |
2 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/c/c_api.cc b/tensorflow/c/c_api.cc index 5c218d3f25..a3003953a3 100644 --- a/tensorflow/c/c_api.cc +++ b/tensorflow/c/c_api.cc @@ -963,6 +963,7 @@ TF_DEVICELIST_METHOD(const char*, TF_DeviceListName, name().c_str(), nullptr); TF_DEVICELIST_METHOD(const char*, TF_DeviceListType, device_type().c_str(), nullptr); TF_DEVICELIST_METHOD(int64_t, TF_DeviceListMemoryBytes, memory_limit(), -1); +TF_DEVICELIST_METHOD(uint64_t, TF_DeviceListIncarnation, incarnation(), 0); #undef TF_DEVICELIST_METHOD diff --git a/tensorflow/c/c_api.h b/tensorflow/c/c_api.h index 1eb75ef11f..fddc09d45e 100644 --- a/tensorflow/c/c_api.h +++ b/tensorflow/c/c_api.h @@ -1521,6 +1521,13 @@ TF_CAPI_EXPORT extern const char* TF_DeviceListType(const TF_DeviceList* list, TF_CAPI_EXPORT extern int64_t TF_DeviceListMemoryBytes( const TF_DeviceList* list, int index, TF_Status* status); +// Retrieve the incarnation number of a given device. +// +// If index is out of bounds, an error code will be set in the status object, +// and 0 will be returned. +TF_CAPI_EXPORT extern uint64_t TF_DeviceListIncarnation( + const TF_DeviceList* list, int index, TF_Status* status); + // -------------------------------------------------------------------------- // Load plugins containing custom ops and kernels |