| /external/tensorflow/tensorflow/core/framework/ |
| D | log_memory.proto | 13 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
|
| D | log_memory.cc | 41 void LogMemory::RecordStep(const int64 step_id, const string& handle) { in RecordStep()
49 const int64 step_id, in RecordTensorAllocation()
67 const int64 step_id, const int index, in RecordTensorOutput()
78 const int64 step_id, size_t num_bytes, in RecordRawAllocation()
91 const int64 step_id, void* ptr, in RecordRawDeallocation()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | partial_run_mgr_test.cc | 27 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
|
| D | partial_run_mgr.cc | 29 bool PartialRunMgr::FindOrCreate(int step_id, in FindOrCreate()
45 void PartialRunMgr::ExecutorDone(int step_id, const Status& executor_status) { in ExecutorDone()
72 void PartialRunMgr::PartialRunDone(int step_id, StatusCallback done, in PartialRunDone()
|
| D | worker_cache_logger.cc | 59 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()
|
| D | rpc_collective_executor_mgr.cc | 48 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()
|
| D | base_rendezvous_mgr.cc | 53 RemoteRendezvous* BaseRendezvousMgr::Find(int64 step_id) { in Find()
57 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
68 void BaseRendezvousMgr::RecvLocalAsync(int64 step_id, in RecvLocalAsync()
85 Status BaseRendezvousMgr::RecvLocal(int64 step_id, in RecvLocal()
104 void BaseRendezvousMgr::Cleanup(int64 step_id) { in Cleanup()
133 BaseRemoteRendezvous::BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id) in BaseRemoteRendezvous()
|
| D | collective_rma_distributed.h | 29 int64 step_id) in CollectiveRemoteAccessDistributed()
|
| D | graph_mgr.cc | 339 Status GraphMgr::SendInputs(const int64 step_id, const NamedTensors& in) { in SendInputs()
355 Status GraphMgr::RecvOutputs(const int64 step_id, NamedTensors* out) { in RecvOutputs()
368 void GraphMgr::RecvOutputsAsync(const int64 step_id, NamedTensors* out, in RecvOutputsAsync()
391 void GraphMgr::ExecuteAsync(const string& handle, const int64 step_id, in ExecuteAsync()
468 void GraphMgr::StartParallelExecutors(const string& handle, int64 step_id, in StartParallelExecutors()
|
| D | worker_cache.h | 83 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { return false; } in RetrieveLogs()
|
| D | worker.cc | 101 void Worker::AbortStep(int64 step_id) { in AbortStep()
157 const int64 step_id = request->step_id(); in DoRunGraph() local
266 const int64 step_id = request->step_id(); in DoPartialRunGraph() local
349 const int64 step_id = request->step_id(); in CleanupGraphAsync() local
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.h | 72 void Init(const Rendezvous::ParsedKey& parsed, const int64 step_id, in Init()
106 void Init(const Rendezvous::ParsedKey& parsed, const int64 step_id) { in Init()
118 MPIRemoteRendezvous(const WorkerEnv* env, int64 step_id, const MPIUtils* util, in MPIRemoteRendezvous()
148 void QueueRequest(std::string key, int64 step_id, in QueueRequest()
194 void GetRecvCall(const int64 step_id, const std::string& key, in GetRecvCall()
206 void RemoveRecvCall(const int64 step_id, const std::string& key) { in RemoveRecvCall()
|
| D | mpi_msg.proto | 14 int64 step_id = 4; field
|
| D | mpi_rendezvous_mgr.cc | 57 BaseRemoteRendezvous* MPIRendezvousMgr::Create(int64 step_id, in Create()
155 const int64 step_id = request.step_id(); in AddRequest() local
291 const int64 step_id = mRes.step_id(); in MPIBackgroundThread() local
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | verbs_util.cc | 28 string VerbsUtil::AppendStepidToKey(const string& key, int64 step_id) { in AppendStepidToKey()
34 int64& step_id) { in GetKeyAndStepId()
|
| D | rdma_rendezvous_mgr.cc | 32 RdmaRemoteRendezvous(const WorkerEnv* env, int64 step_id, RdmaMgr* rdma_mgr) in RdmaRemoteRendezvous()
85 BaseRemoteRendezvous* RdmaRendezvousMgr::Create(int64 step_id, in Create()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 42 CollectiveExecutor* CollectiveExecutorMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
58 CollectiveExecutor* CollectiveExecutorMgr::Create(int64 step_id) { in Create()
65 void CollectiveExecutorMgr::Cleanup(int64 step_id) { in Cleanup()
|
| D | test_collective_executor_mgr.h | 62 CollectiveExecutor* FindOrCreate(int64 step_id) override { in FindOrCreate()
76 void Cleanup(int64 step_id) override { in Cleanup()
110 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId()
|
| D | scoped_allocator_mgr.cc | 130 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()
|
| D | collective_executor_mgr.h | 56 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 87 const int64 step_id = 123; in TEST_F() local
112 const int64 step_id = 123; in TEST_F() local
126 const int64 step_id = 321; in TEST_F() local
146 const int64 step_id = 123; in TEST_F() local
172 const int64 step_id = 123; in TEST_F() local
|
| D | rpc_rendezvous_mgr.cc | 42 RpcRemoteRendezvous(const WorkerEnv* env, int64 step_id) in RpcRemoteRendezvous()
61 void Init(WorkerInterface* wi, int64 step_id, StringPiece key, in Init()
293 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64 step_id, in Create()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 36 RecvBufCall(int64 step_id, const string& peer_device, const string& peer_task, in RecvBufCall()
70 int64 step_id, in CollectiveRemoteAccessDistributed()
150 CollectiveExecutor* GdrCollectiveExecutorMgr::Create(int64 step_id) { in Create()
|
| D | gdr_rendezvous_mgr.cc | 43 const Rendezvous::Args& recv_args, int64 step_id, in GdrRecvTensorCall()
123 GdrRemoteRendezvous(const WorkerEnv* env, int64 step_id, in GdrRemoteRendezvous()
211 BaseRemoteRendezvous* GdrRendezvousMgr::Create(int64 step_id, in Create()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 232 int64 step_id = 2; field
295 int64 step_id = 1; field
312 int64 step_id = 1; field
388 int64 step_id = 1; field
447 int64 step_id = 1; field
|