| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| rpc_rendezvous_mgr.h | 29 // until the tensor is received. Each global unique "step_id"
50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| grpc_worker_service.cc | 374 const int64 step_id = request->step_id(); local
376 TRACEPRINTF("RecvTensor: %lld %s", step_id, key.c_str());
392 opts->SetCancelCallback([this, step_id]() { AbortStep(step_id); });
394 step_id, parsed,
454 for (const auto& step_id : request->fetch_step_id()) {
455 session_mgr->RetrieveLogs(step_id, response);
|
| grpc_remote_worker.cc | 130 int64 step_id = request->step_id();
157 logger_->RecordRecvTensor(step_id, send_start_usec, end_usec,
|
| grpc_worker_cache.cc | 83 bool RetrieveLogs(int64 step_id, StepStats* ss) override {
84 return logger_.RetrieveLogs(step_id, ss);
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_rendezvous_mgr.h | 32 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_session.cc | 130 void RetrieveLogs(int64 step_id, StepStats* ss) {
132 worker_cache_->RetrieveLogs(step_id, ss);
136 req.add_fetch_step_id(step_id);
145 [step_id, ss, resp, &scoped_mu, &waiting_for,
151 if (step_id != lss.step_id()) {
152 LOG(ERROR) << "Wrong step_id in LoggingResponse";
176 Status RunPartitions(const MasterEnv* env, int64 step_id,
182 // Calls workers to cleanup states for the step "step_id". Calls
184 void CleanupPartitionsAsync(int64 step_id, StatusCallback done)
1336 uint64 step_id = (random::New64() & ((1uLL << 56) - 1)) | (1uLL << 56); local
1610 const uint64 step_id = (random::New64() & ((1uLL << 56) - 1)) | (1uLL << 56); local
[all...] |
| graph_mgr.cc | 335 Status GraphMgr::SendInputs(const int64 step_id, const NamedTensors& in) {
336 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id);
351 Status GraphMgr::RecvOutputs(const int64 step_id, NamedTensors* out) {
352 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id);
358 void GraphMgr::RecvOutputsAsync(const int64 step_id, NamedTensors* out,
360 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id);
381 void GraphMgr::ExecuteAsync(const string& handle, const int64 step_id,
415 RemoteRendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id);
438 StartParallelExecutors(handle, step_id, item, rendezvous, collector,
447 void GraphMgr::StartParallelExecutors(const string& handle, int64 step_id,
[all...] |
| worker_cache_wrapper.h | 82 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) {
83 return wrapped_->RetrieveLogs(step_id, ss);
|
| worker_session.cc | 70 bool RetrieveLogs(int64 step_id, StepStats* ss) override {
71 return wrapped_->RetrieveLogs(step_id, ss);
|
| message_wrappers.cc | 297 int64 InMemoryRunGraphRequest::step_id() const { return step_id_; } function in class:tensorflow::InMemoryRunGraphRequest
299 void InMemoryRunGraphRequest::set_step_id(int64 step_id) { step_id_ = step_id; }
368 proto_version_->set_step_id(step_id());
400 int64 MutableProtoRunGraphRequest::step_id() const { function in class:tensorflow::MutableProtoRunGraphRequest
401 return request_.step_id();
404 void MutableProtoRunGraphRequest::set_step_id(int64 step_id) {
405 request_.set_step_id(step_id);
495 int64 ProtoRunGraphRequest::step_id() const { return request_->step_id(); } function in class:tensorflow::ProtoRunGraphRequest
[all...] |
| message_wrappers.h | 255 // The master generates a global unique `step_id` to distinguish
257 // (e.g., send/recv ops) with each other using `step_id` to
259 virtual int64 step_id() const = 0;
297 virtual void set_step_id(int64 step_id) = 0;
317 int64 step_id() const override;
332 void set_step_id(int64 step_id) override;
368 int64 step_id() const override;
383 void set_step_id(int64 step_id) override;
404 int64 step_id() const override;
|
| session_mgr.h | 62 void RetrieveLogs(tensorflow::int64 step_id, LoggingResponse* response);
|
| worker_cache.h | 81 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { return false; }
|
| cluster_function_library_runtime.cc | 182 const uint64 step_id = (random::New64() & ((1uLL << 56) - 1)) | (1uLL << 56); local
183 req.set_step_id(step_id);
|
| master_session.h | 161 uint64 step_id; member in struct:tensorflow::MasterSession::RunState
169 const uint64 step_id, const int64 count);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| captured_function.cc | 197 f_opts.step_id = CapturedFunction::generate_step_id();
198 ScopedStepContainer step_container(f_opts.step_id, [ctx](const string& name) {
231 f_opts.step_id = CapturedFunction::generate_step_id();
232 ScopedStepContainer step_container(f_opts.step_id, [ctx](const string& name) {
276 f_opts.step_id = CapturedFunction::generate_step_id();
279 f_opts.step_id, [resource_mgr](const string& name) {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| function_ops.cc | 76 opts.step_id = ctx->step_id();
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| executor.h | 59 // "step_id" is a process-wide unique identifier for the step being
60 // run. Executors on different devices may receive the same step_id
62 // step_id is used for tracking resource usage of a given step.
84 int64 step_id = 0; member in struct:tensorflow::Executor::Args
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_event_mgr.h | 62 // operation and step_id are only populated when
65 int64 step_id; member in struct:tensorflow::EventMgr::BufRec
123 iu.bufrec.step_id, iu.bufrec.buf,
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| worker.proto | 209 // The master generates a global unique `step_id` to distinguish
211 // (e.g., send/recv ops) with each other using `step_id` to
213 int64 step_id = 2;
276 int64 step_id = 1;
291 // REQUIRED: This must eventually correspond to the `step_id` passed
293 int64 step_id = 1;
366 int64 step_id = 1;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| function_ops.cc | 244 opts.step_id = ctx->step_id();
318 opts.step_id = ctx->step_id();
|
| critical_section.cc | 183 opts_.step_id = CapturedFunction::generate_step_id();
185 new ScopedStepContainer(opts_.step_id, [this](const string& name) {
|
| functional_ops.cc | 100 opts->step_id = ctx->step_id();
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| patch_notes_verbs_with_0_copies.md | 52 | type | name_size | name | step_id | request_index | remote_addr | rkey | is_dead | data_type | tensor_shape | tensor_bytes |
59 * step_id - Step ID.
70 * step_id - Step ID.
|
| README.md | 136 * **class RdmaRemoteRendezvous** - A derived class of BaseRemoteRendezvous. This class is the back end for "send" and "recv" ops. When the sendrecv_op wants to send or receive a tensor, it calls the rendezvous' "send" and "recv" functions respectively. Rendezvous are identified by "step_id", a random number, so that tensors for different iterations don't get mixed up.
140 | type | name_size | name | step_id | request_index | remote_addr/checksum | rkey | is_dead | data_type | tensor_shape | tensor_bytes | error_status |
147 * step_id - Step ID.
159 * step_id - Step ID.
165 * step_id - Step ID.
|
