| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator_mgr.cc | 27 << " step_id_=" << step_id_ << " scope_id=" << scope_id; in AddScopedAllocator()
43 VLOG(2) << " container " << this << " step_id " << step_id_; in AddScopedAllocator()
67 << " in container for step " << step_id_ << " on " in GetAllocator()
74 VLOG(2) << "GetInstance " << scope_id << " step " << step_id_ << " on " in GetInstance()
82 << step_id_ << " on " << mgr_->device_name(); in GetInstance()
88 << step_id_ << " on " << mgr_->device_name(); in Drop()
100 VLOG(2) << "~ScopedAllocatorContainer " << this << " step " << step_id_ in ~ScopedAllocatorContainer()
|
| D | scoped_allocator_mgr_test.cc | 40 VLOG(2) << "Adding ScopedAllocator step_id " << step_id_ << " scope_id " in AddScopedAllocator()
43 return sam_.AddScopedAllocator(backing_tensor_, step_id_, scope_id, in AddScopedAllocator()
57 ScopedAllocatorContainer* sac = sam_.GetContainer(step_id_); in SaveInstances()
90 const int step_id_ = 101; member in tensorflow::__anondcf30ea50111::ScopedAllocatorMgrTest
148 ScopedAllocatorContainer* sac = sam_.GetContainer(step_id_); in TEST_F()
|
| D | collective_rma_local.h | 34 step_id_(step_id) {} in CollectiveRemoteAccessLocal()
78 int64 step_id_; variable
|
| D | buf_rendezvous.h | 45 : step_id_(step_id), dev_mgr_(dev_mgr) {} in BufRendezvous()
123 const uint64 step_id_;
|
| D | scoped_allocator_mgr.h | 49 : mgr_(mgr), step_id_(step_id) {} in ScopedAllocatorContainer()
54 int64 step_id_; variable
|
| D | base_collective_executor.h | 103 step_id_(step_id), in BaseCollectiveExecutor()
139 const int64 step_id_;
|
| D | simple_propagator_state.cc | 34 step_id_(step_id), in SimplePropagatorState()
62 "ExecutorPropagateOutputs#", "id=", step_id_, in PropagateOutputs()
|
| D | executor.cc | 346 int64 step_id_; member in tensorflow::__anonce8950920111::ExecutorState
396 step_id_(args.step_id), in ExecutorState()
416 propagator_(immutable_state, step_id_, vlog_), in ExecutorState()
612 << step_id_ << " " << SummarizeNodeDef(state->item->kernel->def()) in ProcessAsync()
676 {{"id", step_id_}, {"iter_num", tagged_node.get_iter_num()}}); in Process()
678 profiler::ContextType::kTfExecutor, step_id_, in Process()
689 params.step_id = step_id_; in Process()
1231 int64 step_id = step_id_; in Finish()
|
| D | simple_propagator_state.h | 164 const int64 step_id_; variable
|
| D | collective_rma_local.cc | 104 << " step_id_=" << step_id_; in PostToPeer()
|
| D | buf_rendezvous.cc | 252 " step_id=", step_id_, " current contents:"); in LogContents()
|
| D | permuter_test.cc | 374 op_params.step_id = parent_->step_id_; in DoPermute()
428 int64 step_id_ = kStepId; member in tensorflow::__anonfb3b1ff60111::PermuterTest
|
| D | propagator_state.h | 495 const int64 step_id_; variable
|
| D | propagator_state.cc | 31 step_id_(step_id), in PropagatorState()
71 "ExecutorPropagateOutputs#", "id=", step_id_, in PropagateOutputs()
|
| D | base_collective_executor.cc | 317 col_params, exec_key, step_id_, input, output); in ExecuteAsync()
|
| D | hierarchical_tree_broadcaster_test.cc | 632 op_params.step_id = parent_->step_id_; in DoBroadcast()
710 int64 step_id_ = kStepId; member in tensorflow::__anon04765bf90111::HierarchicalTreeBroadcasterTest
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 138 step_id_, "dynamic", to_tensor->dtype(), &to_tensor->shape()); in RecvFromPeer()
172 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | base_rendezvous_mgr.cc | 121 step_id_(step_id), in BaseRemoteRendezvous()
262 ScopedMemoryDebugAnnotation op_annotation("SameWorkerRecvDone", step_id_, in SameWorkerRecvDone()
312 ScopedMemoryDebugAnnotation op_annotation("RecvAsync", step_id_); in RecvAsync()
|
| D | base_rendezvous_mgr.h | 172 const int64 step_id_; variable
|
| D | message_wrappers.cc | 313 int64 InMemoryRunGraphRequest::step_id() const { return step_id_; } in step_id()
315 void InMemoryRunGraphRequest::set_step_id(int64 step_id) { step_id_ = step_id; } in set_step_id()
|
| D | message_wrappers.h | 372 int64 step_id_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 127 step_id_ = context->step_id(); in Reinitialize()
159 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
166 LogMemory::RecordRawDeallocation(operation_, step_id_, buffer, allocator_, in deallocate()
170 new AsyncFreeData(allocator_, buffer, operation_, step_id_); in deallocate()
194 : allocator_(a), address_(p), operation_(o), step_id_(s) {} in AsyncFreeData()
198 const int64 step_id_; member
209 LogMemory::RecordRawDeallocation(data->operation_, data->step_id_, in asyncFree()
217 int64 step_id_; in asyncFree() member in tensorflow::EigenGpuStreamDevice
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_mgr.h | 97 : step_id_(step_id), in ScopedStepContainer()
105 : step_id_(step_id), in ScopedStepContainer()
146 int64 StepId() const { return step_id_; } in StepId()
149 const int64 step_id_;
|
| D | run_handler.cc | 731 int64 step_id() const { return step_id_; } in step_id()
760 int64 step_id_; member in tensorflow::RunHandler::Impl
1070 step_id_ = step_id; in Reset()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 272 call->Init(rwi, step_id_, parsed.FullKey(), recv_args.alloc_attrs, dst_device, in RecvFromRemoteAsync()
|