Searched refs:GetContextId (Results 1 – 17 of 17) sorted by relevance
31 if (ctx->GetContextId() != context_id) { in DestroyRemoteTensorHandle()107 context_id_(ctx->GetContextId()), in RemoteTensorHandleData()
149 request.set_context_id(ctx_->GetContextId()); in StartSend()215 uint64 context_id = ctx_->GetContextId(); in RunRemoteRecv()366 uint64 context_id = ctx_->GetContextId(); in StartSendPackedHandle()419 uint64 context_id = ctx_->GetContextId(); in StartRemoteSendTensor()
98 uint64 GetContextId() const { return context_id_; } in GetContextId() function
61 traceme_context_id = schedule_activity.GetContextId()]() { in ComputeAsync()
176 virtual CUptiResult GetContextId(CUcontext context, uint32_t* context_id) = 0;
169 CUptiResult GetContextId(CUcontext context, uint32_t* context_id) override;
236 CUptiResult CuptiWrapper::GetContextId(CUcontext context, in GetContextId() function in tensorflow::profiler::CuptiWrapper
1087 cupti_interface_->GetContextId(context, &context_id)); in GetContextInfo()
627 request->set_context_id(GetContextId()); in MaybeRegisterFunctionRemotely()662 uint64 context_id = GetContextId(); in RegisterExistingFunctionsOnRemoteWorkers()813 request.set_context_id(GetContextId()); in SyncExecutors()1048 uint64 EagerContext::GetContextId() const { in GetContextId() function in tensorflow::EagerContext
347 uint64 GetContextId() const;
446 tensorflow::uint64 context_id = context->GetContextId(); in UpdateContextWithServerDef()
780 if (remote_mirror->second.context_view_id() >= ctx->GetContextId()) { in AddUnshapedRemoteMirror()
833 uint64 context_id = ctx.GetContextId(); in EagerRemoteExecute()
328 context_id = producer.GetContextId()]() { in ExecuteAsync()
62 return context->GetContextId(); in TFE_GetContextId()
215 } else if (context->GetContextId() == in TFE_ContextUpdateServerDef()
616 collective->trace_context = traceme.GetContextId(); in RunCollective()