| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_worker.cc | 58 const int64 step_id = request->step_id(); local
60 TRACEPRINTF("RecvTensor: %lld %s", step_id, key.c_str());
76 opts->SetCancelCallback([this, step_id]() { AbortStep(step_id); });
79 step_id, parsed,
160 env_->collective_executor_mgr->FindOrCreate(request->step_id()), true);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| rpc_rendezvous_mgr_test.cc | 87 const int64 step_id = 123; local
92 RemoteRendezvous* rendez = rmgr_.Find(step_id);
101 TF_ASSERT_OK(rmgr_.RecvLocal(step_id, key, &val, &val_dead));
104 rmgr_.Cleanup(step_id);
112 const int64 step_id = 123; local
113 RemoteRendezvous* rendez = rmgr_.Find(step_id);
126 const int64 step_id = 321; local
127 RemoteRendezvous* rendez = rmgr_.Find(step_id);
129 SchedClosure([this, step_id]() {
131 rmgr_.Cleanup(step_id);
146 const int64 step_id = 123; local
172 const int64 step_id = 123; local
[all...] |
| grpc_worker_service.cc | 455 const int64 step_id = request->step_id(); local
457 TRACEPRINTF("RecvTensor: %lld %s", step_id, key.c_str());
476 [step_id]() { LOG(WARNING) << "RecvTensor cancelled for " << step_id; });
478 step_id, parsed,
565 env_->collective_executor_mgr->FindOrCreate(request->step_id()), true);
639 for (const auto& step_id : request->fetch_step_id()) {
640 session_mgr->RetrieveLogs(step_id, response);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| rpc_collective_executor_mgr.cc | 48 CollectiveExecutor* RpcCollectiveExecutorMgr::Create(int64 step_id) {
51 worker_cache_, step_id);
52 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_,
61 int64 step_id = random::New64(); local
63 step_id &= kStepIdMask;
64 return step_id;
155 void RpcCollectiveExecutorMgr::RetireStepId(int64 graph_key, int64 step_id) {
159 if (step_id == it->second->next_step_id_) {
|
| cluster_function_library_runtime.cc | 189 const uint64 step_id = (random::New64() & ((1uLL << 56) - 1)) | (1uLL << 56); local
190 req->set_step_id(step_id);
|
| partial_run_mgr_test.cc | 27 int step_id = 1; local
29 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager);
36 int step_id = 1; local
38 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager);
41 partial_run_mgr.FindOrCreate(step_id, &found_cancellation_manager);
48 int step_id = 1; local
50 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager);
62 int step_id = 1; local
64 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager);
68 step_id, [&called](Status status) { called++; }, Status::OK())
109 int step_id = 1; local
125 int step_id = 1; local
[all...] |
| master_session.h | 179 uint64 step_id; member in struct:tensorflow::MasterSession::RunState
188 const uint64 step_id, const int64 count);
231 const RunOptions& run_options, uint64 step_id,
242 Status PostRunCleanup(MasterSession::ReffedClientGraph* rcg, uint64 step_id,
|
| worker.cc | 101 void Worker::AbortStep(int64 step_id) {
102 Rendezvous* rendez = env_->rendezvous_mgr->Find(step_id);
103 SchedNonBlockingClosureAfter(1000000, [rendez, step_id]() {
107 rendez->StartAbort(errors::Aborted("Step ", step_id,
157 const int64 step_id = request->step_id(); local
158 TRACEPRINTF("RunGraph: %lld", step_id);
208 opts->SetCancelCallback([this, cm, step_id]() {
210 AbortStep(step_id);
226 request->graph_handle(), step_id, session.get(), request->exec_opts()
266 const int64 step_id = request->step_id(); local
349 const int64 step_id = request->step_id(); local
[all...] |
| message_wrappers.cc | 315 int64 InMemoryRunGraphRequest::step_id() const { return step_id_; } function in class:tensorflow::InMemoryRunGraphRequest
317 void InMemoryRunGraphRequest::set_step_id(int64 step_id) { step_id_ = step_id; }
402 proto_version_->set_step_id(step_id());
443 int64 MutableProtoRunGraphRequest::step_id() const { function in class:tensorflow::MutableProtoRunGraphRequest
444 return request_.step_id();
447 void MutableProtoRunGraphRequest::set_step_id(int64 step_id) {
448 request_.set_step_id(step_id);
554 int64 ProtoRunGraphRequest::step_id() const { return request_->step_id(); } function in class:tensorflow::ProtoRunGraphRequest
[all...] |
| master_session.cc | 155 void RetrieveLogs(int64 step_id, StepStats* ss) {
157 worker_cache_->RetrieveLogs(step_id, ss);
161 req.add_fetch_step_id(step_id);
170 [step_id, ss, resp, &scoped_mu, &all_done](const Status& s) {
175 if (step_id != lss.step_id()) {
176 LOG(ERROR) << "Wrong step_id in LoggingResponse";
200 Status RunPartitions(const MasterEnv* env, int64 step_id,
205 Status RunPartitions(const MasterEnv* env, int64 step_id,
210 // Calls workers to cleanup states for the step "step_id". Call
1483 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); local
1881 uint64 step_id = NewStepId(rcg->collective_graph_key()); local
1946 const uint64 step_id = NewStepId(rcg->collective_graph_key()); local
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_event_mgr.h | 78 // operation and step_id are only populated when
81 int64 step_id; member in struct:tensorflow::EventMgr::BufRec
139 iu.bufrec.step_id, iu.bufrec.buf,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| scoped_allocator_ops_test.cc | 59 allocator = device_->GetScopedAllocator(attr, context_->step_id());
100 int64 step_id = 10; local
101 Status s = sam->AddScopedAllocator(**backing_tensor, step_id, id,
106 ScopedAllocatorContainer* sac = sam->GetContainer(step_id);
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| mpi_rendezvous_mgr.cc | 57 BaseRemoteRendezvous* MPIRendezvousMgr::Create(int64 step_id,
59 return new MPIRemoteRendezvous(worker_env, step_id, mpiutils_, this);
155 const int64 step_id = request.step_id(); local
199 [this, parsed, step_id, send_cb](
205 << " step: " << step_id
211 << " @ step: " << step_id << std::endl;
214 mpi_send_call->Init(parsed, step_id, is_dead);
271 worker_env_2->compute_pool->Schedule([this, step_id, parsed, done_cb]() {
272 this->RecvLocalAsync(step_id, parsed, done_cb)
291 const int64 step_id = mRes.step_id(); local
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| executor.h | 60 // "step_id" is a process-wide unique identifier for the step being
61 // run. Executors on different devices may receive the same step_id
63 // step_id is used for tracking resource usage of a given step.
85 int64 step_id = 0; member in struct:tensorflow::Executor::Args
|
| direct_session.cc | 462 Status DirectSession::RunInternal(int64 step_id, const RunOptions& run_options,
467 string session_id_meta = strings::StrCat("SessionRun #id=", step_id, "#");
476 run_options.debug_options().global_step(), step_id,
481 RunState run_state(step_id, &devices_);
511 collective_executor_mgr_->FindOrCreate(step_id), true /*inherit_ref*/));
527 args.step_id = step_id;
782 const int64 step_id = step_id_counter_.fetch_add(1); local
785 LogMemory::RecordStep(step_id, run_state_args.handle);
788 TF_RETURN_IF_ERROR(RunInternal(step_id, run_options, &call_frame
1793 const int64 step_id = step_id_counter_.fetch_add(1); local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| collective.h | 186 // Used with a distributed implementation to coordinate step_id
192 // Refresh the local per-graph_key step_id sequence from collective
197 // Returns the step_id that should be used for initiating a new execution
198 // on the specified graph. May return the same step_id multiple times if
205 virtual void RetireStepId(int64 graph_key, int64 step_id) = 0;
216 virtual CollectiveExecutor* FindOrCreate(int64 step_id) = 0;
218 // If there is a CollectiveExecutor for step_id, remove it from the
220 virtual void Cleanup(int64 step_id) = 0;
344 int64 step_id, const Tensor* input, Tensor* output);
354 const int64 step_id; member in class:tensorflow::CollectiveContext
[all...] |
| function.h | 616 int64 step_id = 0; member in struct:tensorflow::FunctionLibraryRuntime::Options
[all...] |
| op_kernel.h | 581 int64 step_id = 0; member in struct:tensorflow::OpKernelContext::Params
693 int64 step_id() const { return params_->step_id; } function in class:tensorflow::OpKernelContext
[all...] |
