| /external/tensorflow/tensorflow/contrib/verbs/ |
| H A D | rdma_rendezvous_mgr.cc | 67 string key_with_step_id = VerbsUtil::AppendStepidToKey(key, step_id_);
78 rc->InsertTensorRequest(key, step_id_, dst_dev, recv_args, done);
|
| H A D | rdma.cc | 484 RDMA_LOG(1) << "Step 0x" << std::hex << rm.step_id_ << std::dec
1005 rm_.step_id_, parsed,
1175 rm.step_id_ = rm_.step_id_;
1182 RDMA_LOG(1) << "Step 0x" << std::hex << rm.step_id_ << std::dec
1221 RDMA_LOG(1) << "Step 0x" << std::hex << rm_.step_id_ << std::dec
1238 rm.step_id_ = rm_.step_id_;
1241 LOG(ERROR) << "Step 0x" << std::hex << rm.step_id_ << std::dec
1310 memcpy(&message[kStepIdStartIndex], &rm.step_id_, sizeo
[all...] |
| H A D | rdma.h | 90 int64 step_id_; member in struct:tensorflow::RdmaMessage
118 kStepIdStartIndex + sizeof(step_id_);
291 int64 step_id_; member in class:tensorflow::RdmaTensorRequest
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| H A D | mpi_rendezvous_mgr.cc | 69 << " @ step: " << step_id_;
85 rendezvous_call->Init(parsed, step_id_);
103 const int64 temp1 = step_id_;
139 mgr->QueueRequest(parsed.FullKey().ToString(), step_id_,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | base_rendezvous_mgr.h | 175 const int64 step_id_; member in class:tensorflow::BaseRemoteRendezvous
|
| H A D | base_rendezvous_mgr.cc | 135 step_id_(step_id),
|
| H A D | message_wrappers.cc | 297 int64 InMemoryRunGraphRequest::step_id() const { return step_id_; }
299 void InMemoryRunGraphRequest::set_step_id(int64 step_id) { step_id_ = step_id; }
|
| H A D | message_wrappers.h | 345 int64 step_id_; member in class:tensorflow::InMemoryRunGraphRequest
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device.cc | 99 step_id_ = context->step_id();
130 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret,
137 LogMemory::RecordRawDeallocation(operation_, step_id_, buffer, allocator_,
141 new AsyncFreeData(allocator_, buffer, operation_, step_id_);
160 : allocator_(a), address_(p), operation_(o), step_id_(s) {}
164 const int64 step_id_; member in struct:tensorflow::EigenCudaStreamDevice::AsyncFreeData
171 LogMemory::RecordRawDeallocation(data->operation_, data->step_id_,
179 int64 step_id_; member in class:tensorflow::EigenCudaStreamDevice
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| H A D | gdr_rendezvous_mgr.cc | 168 recv_args, step_id_, parsed.FullKey());
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | rpc_rendezvous_mgr.cc | 241 call->Init(rwi, step_id_, parsed.FullKey(), recv_args.alloc_attrs, dst_device,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | executor.cc | 1173 int64 step_id_; member in class:tensorflow::__anon26154::ExecutorState
1296 step_id_(args.step_id),
1502 params.step_id = step_id_;
1619 << " step " << step_id_ << " "
|