Searched refs:GetContextViewId (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 79 handle->SetRemoteShape(shape, remote_device_, ctx_->GetContextViewId())); in TEST_F()
118 handle->HasRemoteMirror(remote_device_, ctx_->GetContextViewId())); in TEST_F()
123 handle->HasRemoteMirror(remote_device_, ctx_->GetContextViewId())); in TEST_F()
126 ctx_->GetContextViewId()) in TEST_F()
140 ctx_->GetContextViewId())); in TEST_F()
150 ctx_->GetContextViewId())); in TEST_F()
|
| D | eager_service_impl.cc | 212 context_it->second->Context()->GetContextViewId()) { in CreateContext()
342 if (request->context_view_id() != ctx->GetContextViewId() + 1) { in UpdateContext()
345 request->context_id(), "> currently at view #", ctx->GetContextViewId(), in UpdateContext()
625 response->set_context_view_id(ctx->GetContextViewId()); in KeepAlive()
640 if (request->context_view_id() < context->Context()->GetContextViewId()) { in CloseContext()
645 << context->Context()->GetContextViewId() << "."; in CloseContext()
|
| D | remote_copy_node.cc | 220 uint64 context_view_id = ctx_->GetContextViewId(); in RunRemoteRecv()
356 const uint64 context_view_id = ctx_->GetContextViewId(); in StartSendPackedHandle()
424 uint64 context_view_id = ctx_->GetContextViewId(); in StartRemoteSendTensor()
509 uint64 context_view_id = ctx_->GetContextViewId(); in Abort()
|
| D | remote_tensor_handle_data.cc | 108 context_view_id_(ctx->GetContextViewId()), in RemoteTensorHandleData()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 209 context()->GetContextViewId())); in TEST_F()
212 context()->GetContextViewId())); in TEST_F()
399 shape, d1, context->GetContextViewId(), d2->name())); in TEST_F()
|
| D | execute_node.cc | 24 uint64 context_view_id = ctx->GetContextViewId(); in IsRemote()
|
| D | execute.cc | 892 ctx.GetContextViewId())); in EagerRemoteExecute()
961 ctx.GetContextViewId(), eager_client.get(), op->GetCancellationManager(), in EagerRemoteExecute()
1009 output_device, ctx->GetContextViewId())); in GetKernelOutputs()
1038 ctx->GetContextViewId())); in GetKernelOutputs()
1272 if (h->HasRemoteMirror(device, ctx->GetContextViewId())) { in EagerCopyToDevice()
|
| D | tensor_handle.cc | 413 ctx->GetContextViewId(), is_ready) { in TensorHandle()
801 if (mirror->second.context_view_id() == ctx->GetContextViewId()) { in AddResourceShapeMirror()
811 std::forward_as_tuple(op_id, output_num, ctx->GetContextViewId(), in AddResourceShapeMirror()
|
| D | context_distributed_manager.cc | 447 tensorflow::uint64 context_view_id = context->GetContextViewId(); in UpdateContextWithServerDef()
509 &existing_workers, context_id, context->GetContextViewId(), server_def, in UpdateContextWithServerDef()
|
| D | context.h | 348 uint64 GetContextViewId() const;
|
| D | context.cc | 1053 uint64 EagerContext::GetContextViewId() const { in GetContextViewId() function in tensorflow::EagerContext
1251 CloseRemoteContexts(remove_remote_contexts, context_id, GetContextViewId()); in UpdateRemoteMaster()
|