diff options
Diffstat (limited to 'tensorflow/core/common_runtime/eager/context.h')
-rw-r--r-- | tensorflow/core/common_runtime/eager/context.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/common_runtime/eager/context.h b/tensorflow/core/common_runtime/eager/context.h index 21c5bdf8e9..9835b19511 100644 --- a/tensorflow/core/common_runtime/eager/context.h +++ b/tensorflow/core/common_runtime/eager/context.h @@ -206,6 +206,7 @@ class EagerContext { // Only one of the below is set. std::unique_ptr<DeviceMgr> local_device_manager_; DeviceMgr* local_unowned_device_manager_; + std::unique_ptr<DeviceMgr> remote_device_manager_; // Devices owned by device_manager std::vector<Device*> devices_; @@ -253,7 +254,6 @@ class EagerContext { #ifndef __ANDROID__ void CloseRemoteContexts(); - std::unique_ptr<DeviceMgr> remote_device_manager_; // The server_ is not const since we release it when the context is destroyed. // Therefore the server_ object is not marked as const (even though it should |