Home
last modified time | relevance | path

Searched defs:step_id (Results 1 – 25 of 69) sorted by relevance

123

/external/tensorflow/tensorflow/core/framework/
Dlog_memory.proto13 int64 step_id = 1; field
21 int64 step_id = 1; field
42 int64 step_id = 1; field
57 int64 step_id = 1; field
78 int64 step_id = 1; field
Dlog_memory.cc40 void LogMemory::RecordStep(const int64 step_id, const string& handle) { in RecordStep()
48 const int64 step_id, in RecordTensorAllocation()
66 const int64 step_id, const int index, in RecordTensorOutput()
77 const int64 step_id, size_t num_bytes, in RecordRawAllocation()
90 const int64 step_id, void* ptr, in RecordRawDeallocation()
Dresource_mgr.h94 ScopedStepContainer(const int64 step_id, in ScopedStepContainer()
101 ScopedStepContainer(const int64 step_id, in ScopedStepContainer()
146 const int64 step_id() const { return step_id_; } in step_id() function
/external/tensorflow/tensorflow/core/distributed_runtime/
Dpartial_run_mgr_test.cc27 int step_id = 1; in TEST() local
36 int step_id = 1; in TEST() local
48 int step_id = 1; in TEST() local
62 int step_id = 1; in TEST() local
109 int step_id = 1; in TEST_P() local
125 int step_id = 1; in TEST_P() local
Dpartial_run_mgr.cc22 bool PartialRunMgr::FindOrCreate(int step_id, in FindOrCreate()
40 void PartialRunMgr::ExecutorDone(int step_id, const Status& executor_status) { in ExecutorDone()
67 void PartialRunMgr::PartialRunDone(int step_id, StatusCallback done, in PartialRunDone()
Dworker_cache_logger.cc59 bool WorkerCacheLogger::RetrieveLogs(int64 step_id, StepStats* ss) { in RetrieveLogs()
71 void WorkerCacheLogger::Save(const string& device, int64 step_id, in Save()
85 void WorkerCacheLogger::RecordRecvTensor(int64 step_id, int64 start_usecs, in RecordRecvTensor()
95 void WorkerCacheLogger::RecordDataTransfer(int64 step_id, int64 start_usecs, in RecordDataTransfer()
Drpc_collective_executor_mgr.cc48 CollectiveExecutor* RpcCollectiveExecutorMgr::Create(int64 step_id) { in Create()
61 int64 step_id = random::New64(); in NewRandomStepId() local
155 void RpcCollectiveExecutorMgr::RetireStepId(int64 graph_key, int64 step_id) { in RetireStepId()
Dbase_rendezvous_mgr.cc55 RemoteRendezvous* BaseRendezvousMgr::Find(int64 step_id) { in Find()
59 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
70 void BaseRendezvousMgr::RecvLocalAsync(int64 step_id, in RecvLocalAsync()
84 Status BaseRendezvousMgr::RecvLocal(int64 step_id, in RecvLocal()
103 void BaseRendezvousMgr::Cleanup(int64 step_id) { in Cleanup()
132 BaseRemoteRendezvous::BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id) in BaseRemoteRendezvous()
Dgraph_mgr.cc260 Rendezvous** r) -> Status { in InitItem()
345 Status GraphMgr::SendInputs(const int64 step_id, const NamedTensors& in) { in SendInputs()
364 Status GraphMgr::RecvOutputs(const int64 step_id, NamedTensors* out) { in RecvOutputs()
382 void GraphMgr::RecvOutputsAsync(const int64 step_id, NamedTensors* out, in RecvOutputsAsync()
408 void GraphMgr::ExecuteAsync(const string& handle, const int64 step_id, in ExecuteAsync()
498 const string& handle, int64 step_id, Item* item, Rendezvous* rendezvous, in StartParallelExecutors()
Dcollective_rma_distributed.h31 WorkerCacheInterface* worker_cache, int64 step_id) in CollectiveRemoteAccessDistributed()
Dworker_cache.h85 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { return false; } in RetrieveLogs()
Dworker.cc107 void Worker::AbortStep(int64 step_id) { in AbortStep()
163 const int64 step_id = request->step_id(); in DoRunGraph() local
262 const int64 step_id = request->step_id(); in DoPartialRunGraph() local
351 const int64 step_id = request->step_id(); in CleanupGraphAsync() local
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_executor_mgr.cc44 CollectiveExecutor* CollectiveExecutorMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
60 CollectiveExecutor* CollectiveExecutorMgr::Create(int64 step_id) { in Create()
67 void CollectiveExecutorMgr::Cleanup(int64 step_id) { in Cleanup()
Dtest_collective_executor_mgr.h66 CollectiveExecutor* FindOrCreate(int64 step_id) override { in FindOrCreate()
80 void Cleanup(int64 step_id) override { in Cleanup()
114 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId()
Dscoped_allocator_mgr.cc130 void ScopedAllocatorMgr::Cleanup(int64 step_id) { in Cleanup()
139 ScopedAllocatorContainer* ScopedAllocatorMgr::GetContainer(int64 step_id) { in GetContainer()
154 const Tensor& backing_tensor, int64 step_id, int32 scope_id, in AddScopedAllocator()
Dcollective_executor_mgr.h57 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId()
Dscoped_allocator_mgr.h47 ScopedAllocatorContainer(const ScopedAllocatorMgr* mgr, int64 step_id) in ScopedAllocatorContainer()
Dcollective_rma_local.h32 int64 step_id) in CollectiveRemoteAccessLocal()
Dbuf_rendezvous.h43 explicit BufRendezvous(uint64 step_id, const DeviceMgr* dev_mgr) in BufRendezvous()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc92 const int64 step_id = 123; in TEST_F() local
117 const int64 step_id = 123; in TEST_F() local
131 const int64 step_id = 321; in TEST_F() local
151 const int64 step_id = 123; in TEST_F() local
175 const int64 step_id = 123; in TEST_F() local
201 const int64 step_id = 123; in TEST_F() local
227 const int64 step_id = 123; in TEST_F() local
Dgrpc_response_cache.cc22 bool GrpcResponseCache::QueueRequest(int64 request_id, int64 step_id, in QueueRequest()
101 void GrpcResponseCache::CleanEntriesForStep(int64 step_id) { in CleanEntriesForStep()
Drpc_rendezvous_mgr.cc42 RpcRemoteRendezvous(const WorkerEnv* env, int64 step_id) in RpcRemoteRendezvous()
61 void Init(WorkerInterface* wi, int64 step_id, StringPiece key, in Init()
289 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64 step_id, in Create()
Dgrpc_response_cache.h76 int64 step_id = -1; member
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto238 int64 step_id = 2; field
311 int64 step_id = 1; field
327 int64 step_id = 1; field
416 int64 step_id = 1; field
474 int64 step_id = 1; field
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.cc76 const string NcclBase::NcclCollectiveKey(const string& exec_key, int step_id) { in NcclCollectiveKey()

123