Searched refs:GetContextViewId (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 82 handle->SetRemoteShape(shape, remote_device_, ctx_->GetContextViewId())); in TEST_F()
121 handle->HasRemoteMirror(remote_device_, ctx_->GetContextViewId())); in TEST_F()
126 handle->HasRemoteMirror(remote_device_, ctx_->GetContextViewId())); in TEST_F()
129 ctx_->GetContextViewId()) in TEST_F()
143 ctx_->GetContextViewId())); in TEST_F()
153 ctx_->GetContextViewId())); in TEST_F()
|
| D | eager_service_impl.cc | 217 context_it->second->Context()->GetContextViewId()) { in CreateContext()
378 if (request->context_view_id() != ctx->GetContextViewId() + 1) { in UpdateContext()
381 request->context_id(), "> currently at view #", ctx->GetContextViewId(), in UpdateContext()
664 response->set_context_view_id(ctx->GetContextViewId()); in KeepAlive()
679 if (request->context_view_id() < context->Context()->GetContextViewId()) { in CloseContext()
684 << context->Context()->GetContextViewId() << "."; in CloseContext()
|
| D | remote_copy_node.cc | 228 uint64 context_view_id = ctx_->GetContextViewId(); in RunRemoteRecv()
365 const uint64 context_view_id = ctx_->GetContextViewId(); in StartSendPackedHandle()
434 uint64 context_view_id = ctx_->GetContextViewId(); in StartRemoteSendTensor()
518 uint64 context_view_id = ctx_->GetContextViewId(); in Abort()
|
| D | remote_tensor_handle_data.cc | 109 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 | 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 | execute.cc | 1600 ctx.GetContextViewId())); in EagerRemoteExecute()
1669 ctx.GetContextViewId(), eager_client.get(), op->GetCancellationManager(), in EagerRemoteExecute()
1720 output_device, ctx->GetContextViewId())); in GetKernelOutputs()
1749 ctx->GetContextViewId())); in GetKernelOutputs()
1998 if (h->HasRemoteMirror(device, ctx->GetContextViewId())) { in EagerCopyToDevice()
|
| D | context_distributed_manager.cc | 443 uint64 context_view_id = context->GetContextViewId(); in UpdateContextWithServerDef()
505 &existing_workers, context_id, context->GetContextViewId(), server_def, in UpdateContextWithServerDef()
|
| D | context.cc | 1306 uint64 EagerContext::GetContextViewId() const { in GetContextViewId() function in tensorflow::EagerContext
1508 CloseRemoteContexts(remove_remote_contexts, context_id, GetContextViewId()); in UpdateRemoteMaster()
|
| D | context.h | 397 uint64 GetContextViewId() const;
|