Home
last modified time | relevance | path

Searched refs:context_id_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/lib/
Dconnected_traceme.h92 context_id_ = in trace_me_()
94 return TraceMeEncode({{"_pt", context_type}, {"_p", context_id_}}); in trace_me_()
98 uint64 GetContextId() const { return context_id_; } in GetContextId()
102 uint64 context_id_ = 0; variable
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.h40 : context_id_(context_id), in EagerClusterFunctionLibraryRuntime()
82 const uint64 context_id_;
Deager_service_impl_test.cc708 context_id_ = random::New64(); in FunctionWithRemoteInputsTest()
745 request.set_context_id(context_id_); in Init()
751 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in Init()
761 remote_enqueue_request.set_context_id(context_id_); in Init()
775 context_id_, ctx, device_mgr_.get()); in Init()
801 close_context_request.set_context_id(context_id_); in CheckOutputTensorAndClose()
813 context_id_, RemoteTensorHandleInternal(2, 0), &tensor_handle)); in CheckOutputsAndClose()
827 uint64 context_id_; member in tensorflow::eager::__anon09d935ad0111::FunctionWithRemoteInputsTest
848 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in TEST_F()
909 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in TEST_F()
[all …]
Dcluster_function_library_runtime.cc87 request->set_context_id(context_id_); in Instantiate()
170 request->set_context_id(context_id_); in Run()
273 request->set_context_id(context_id_); in CleanUp()
Dremote_tensor_handle_data.cc107 context_id_(ctx->GetContextId()), in RemoteTensorHandleData()
118 DestroyRemoteTensorHandle(ctx_, remote_task_, context_id_, op_id_, in ~RemoteTensorHandleData()
Dremote_tensor_handle_data.h78 uint64 context_id_; variable
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc115 context_id_ = kInvalidContextId;
429 context_id = context_id_; in CloseAndClearAllRemoteContexts()
431 context_id_ = kInvalidContextId; in CloseAndClearAllRemoteContexts()
1050 return context_id_; in GetContextId()
1238 if (context_id != context_id_) { in UpdateRemoteMaster()
1241 "Request id = ", context_id, " but current id = ", context_id_); in UpdateRemoteMaster()
1309 context_id_ = context_id; in SetMasterContextState()
1401 request->set_context_id(context_id_); in SetMasterContextState()
1445 context_id_ = context_id; in InitializeRemoteWorker()
1480 if (context_id != context_id_) { in UpdateRemoteWorker()
[all …]
Dcontext.h687 uint64 context_id_ TF_GUARDED_BY(remote_state_mu_);