Searched refs:remote_contexts (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/c/eager/ |
D | c_api.cc | 117 tensorflow::gtl::FlatMap<string, tensorflow::uint64>* remote_contexts) { in CreateRemoteContexts() argument 151 remote_contexts->emplace(remote_worker, response.context_id()); in CreateRemoteContexts() 207 tensorflow::gtl::FlatMap<string, tensorflow::uint64> remote_contexts; in UpdateTFE_ContextWithServerDef() local 210 remote_eager_workers.get(), ctx->context.Async(), &remote_contexts)); in UpdateTFE_ContextWithServerDef() 231 std::move(remote_device_mgr), remote_contexts, r, device_mgr, in UpdateTFE_ContextWithServerDef()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | context.cc | 463 const gtl::FlatMap<string, uint64>& remote_contexts, Rendezvous* r, in InitializeRemote() argument 470 remote_contexts_ = remote_contexts; in InitializeRemote()
|
D | context.h | 222 const gtl::FlatMap<string, uint64>& remote_contexts, Rendezvous* r,
|