Home
last modified time | relevance | path

Searched refs:remote_device_mgr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api.cc104 std::unique_ptr<tensorflow::DeviceMgr> remote_device_mgr( in GetAllRemoteDevices() local
109 *device_mgr = std::move(remote_device_mgr); in GetAllRemoteDevices()
196 std::unique_ptr<tensorflow::DeviceMgr> remote_device_mgr; in UpdateTFE_ContextWithServerDef() local
199 &remote_device_mgr)); in UpdateTFE_ContextWithServerDef()
231 std::move(remote_device_mgr), remote_contexts, r, device_mgr, in UpdateTFE_ContextWithServerDef()
370 if (ctx->context.remote_device_mgr()) { in TFE_ContextListDevices()
371 ctx->context.remote_device_mgr()->ListDeviceAttributes(&list->response); in TFE_ContextListDevices()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h176 const tensorflow::DeviceMgr* remote_device_mgr() const { in remote_device_mgr() function
Dexecute.cc246 if (d == nullptr || ctx->remote_device_mgr() == nullptr) return true; in IsLocal()
287 if (ctx->remote_device_mgr() != nullptr) { in FindDeviceFromName()
288 return ctx->remote_device_mgr()->LookupDevice(device_name, device); in FindDeviceFromName()