Home
last modified time | relevance | path

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

12

/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.cc41 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/
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.cc29 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()
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()
Dbase_rendezvous_mgr.cc53 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()
Dgraph_mgr.cc335 Status GraphMgr::SendInputs(const int64 step_id, const NamedTensors& in) { in SendInputs()
351 Status GraphMgr::RecvOutputs(const int64 step_id, NamedTensors* out) { in RecvOutputs()
358 void GraphMgr::RecvOutputsAsync(const int64 step_id, NamedTensors* out, in RecvOutputsAsync()
381 void GraphMgr::ExecuteAsync(const string& handle, const int64 step_id, in ExecuteAsync()
447 void GraphMgr::StartParallelExecutors(const string& handle, int64 step_id, in StartParallelExecutors()
Dworker_cache.h81 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { return false; } in RetrieveLogs()
Dworker.cc81 void Worker::AbortStep(int64 step_id) { in AbortStep()
136 const int64 step_id = request->step_id(); in DoRunGraph() local
209 const int64 step_id = request->step_id(); in DoPartialRunGraph() local
287 const int64 step_id = request->step_id(); in CleanupGraphAsync() local
Dworker_cache_wrapper.h82 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { in RetrieveLogs()
Dworker_session.cc70 bool RetrieveLogs(int64 step_id, StepStats* ss) override { in RetrieveLogs()
Dmaster_session.cc130 void RetrieveLogs(int64 step_id, StepStats* ss) { in RetrieveLogs()
494 const MasterEnv* env, int64 step_id, int64 execution_count, in RunPartitions()
673 CleanupBroadcastHelper(int64 step_id, int num_calls, StatusCallback done) in CleanupBroadcastHelper()
723 int64 step_id, StatusCallback done) { in CleanupPartitionsAsync()
736 void MasterSession::ReffedClientGraph::ProcessStats(int64 step_id, in ProcessStats()
1336 uint64 step_id = (random::New64() & ((1uLL << 56) - 1)) | (1uLL << 56); in PartialRunSetup() local
1610 const uint64 step_id = (random::New64() & ((1uLL << 56) - 1)) | (1uLL << 56); in DoRunWithLocalExecution() local
1707 ReffedClientGraph* rcg, const uint64 step_id, in RunState()
Dmessage_wrappers.cc297 int64 InMemoryRunGraphRequest::step_id() const { return step_id_; } in step_id() function in tensorflow::InMemoryRunGraphRequest
299 void InMemoryRunGraphRequest::set_step_id(int64 step_id) { step_id_ = step_id; } in set_step_id()
400 int64 MutableProtoRunGraphRequest::step_id() const { in step_id() function in tensorflow::MutableProtoRunGraphRequest
404 void MutableProtoRunGraphRequest::set_step_id(int64 step_id) { in set_step_id()
495 int64 ProtoRunGraphRequest::step_id() const { return request_->step_id(); } in step_id() function in tensorflow::ProtoRunGraphRequest
/external/tensorflow/tensorflow/contrib/verbs/
Dverbs_util.cc28 string VerbsUtil::AppendStepidToKey(const string& key, int64 step_id) { in AppendStepidToKey()
34 int64& step_id) { in GetKeyAndStepId()
Drdma_rendezvous_mgr.cc32 RdmaRemoteRendezvous(const WorkerEnv* env, int64 step_id, RdmaMgr* rdma_mgr) in RdmaRemoteRendezvous()
85 BaseRemoteRendezvous* RdmaRendezvousMgr::Create(int64 step_id, in Create()
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.h72 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()
Dmpi_msg.proto14 int64 step_id = 4; field
Dmpi_rendezvous_mgr.cc57 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/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc85 const int64 step_id = 123; in TEST_F() local
110 const int64 step_id = 123; in TEST_F() local
124 const int64 step_id = 321; in TEST_F() local
144 const int64 step_id = 123; in TEST_F() local
170 const int64 step_id = 123; in TEST_F() local
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()
268 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64 step_id, in Create()
Dgrpc_worker_cache.cc83 bool RetrieveLogs(int64 step_id, StepStats* ss) override { in RetrieveLogs()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto213 int64 step_id = 2; field
276 int64 step_id = 1; field
293 int64 step_id = 1; field
366 int64 step_id = 1; field
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_rendezvous_mgr.cc43 const Rendezvous::Args& recv_args, int64 step_id, in GdrRecvTensorCall()
128 GdrRemoteRendezvous(const WorkerEnv* env, int64 step_id, in GdrRemoteRendezvous()
202 BaseRemoteRendezvous* GdrRendezvousMgr::Create(int64 step_id, in Create()
Dgdr_worker.cc59 const int64 step_id = request->step_id(); in GrpcRecvTensorAsync() local
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr.h65 int64 step_id; member

12