| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| base_rendezvous_mgr.h | 175 const int64 step_id_; member in class:tensorflow::BaseRemoteRendezvous
|
| message_wrappers.h | 345 int64 step_id_; member in class:tensorflow::InMemoryRunGraphRequest
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| 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/core/common_runtime/gpu/ |
| 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
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| executor.cc | 1173 int64 step_id_; member in class:tensorflow::__anon39475::ExecutorState
[all...] |
