Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/framework/
Dlog_memory.proto15 int64 step_id = 1; field
23 int64 step_id = 1; field
44 int64 step_id = 1; field
59 int64 step_id = 1; field
80 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()
/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.cc50 CollectiveExecutor* RpcCollectiveExecutorMgr::Create(int64 step_id) { in Create()
64 int64 step_id = random::New64(); in NewRandomStepId() local
158 void RpcCollectiveExecutorMgr::RetireStepId(int64 graph_key, int64 step_id) { in RetireStepId()
Dgraph_mgr.cc152 [this](const int64 step_id) { in InitItem()
352 Status GraphMgr::SendInputs(const int64 step_id, const NamedTensors& in) { in SendInputs()
371 Status GraphMgr::RecvOutputs(const int64 step_id, NamedTensors* out) { in RecvOutputs()
389 void GraphMgr::RecvOutputsAsync(const int64 step_id, NamedTensors* out, in RecvOutputsAsync()
415 void GraphMgr::ExecuteAsync(const string& handle, const int64 step_id, in ExecuteAsync()
512 const string& handle, int64 step_id, Item* item, Rendezvous* rendezvous, in StartParallelExecutors()
Dbase_rendezvous_mgr.cc56 RemoteRendezvous* BaseRendezvousMgr::Find(int64 step_id) { in Find()
60 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
71 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()
119 BaseRemoteRendezvous::BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id) in BaseRemoteRendezvous()
Dworker.cc108 void Worker::AbortStep(int64 step_id) { in AbortStep()
164 const int64 step_id = request->step_id(); in DoRunGraph() local
265 const int64 step_id = request->step_id(); in DoPartialRunGraph() local
354 const int64 step_id = request->step_id(); in CleanupGraphAsync() local
Dcollective_rma_distributed.h32 WorkerCacheInterface* worker_cache, int64 step_id, string task_name) in CollectiveRemoteAccessDistributed()
Dworker_cache.h85 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { return false; } in RetrieveLogs()
Dmaster_session.cc159 void RetrieveLogs(int64 step_id, StepStats* ss) { in RetrieveLogs()
644 const FetchListType& fetches, const MasterEnv* env, int64 step_id, in RunPartitionsHelper()
808 const MasterEnv* env, int64 step_id, int64 execution_count, in RunPartitions()
833 const MasterEnv* env, int64 step_id, int64 execution_count, in RunPartitions()
877 CleanupBroadcastHelper(int64 step_id, int num_calls, StatusCallback done) in CleanupBroadcastHelper()
927 int64 step_id, StatusCallback done) { in CleanupPartitionsAsync()
940 void MasterSession::ReffedClientGraph::ProcessStats(int64 step_id, in ProcessStats()
1575 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId() local
1875 uint64 step_id, int64 count, in FillPerStepState()
1902 uint64 step_id, in PostRunCleanup()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_executor_mgr.cc46 CollectiveExecutor* CollectiveExecutorMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
62 CollectiveExecutor* CollectiveExecutorMgr::Create(int64 step_id) { in Create()
69 void CollectiveExecutorMgr::Cleanup(int64 step_id) { in Cleanup()
Dtest_collective_executor_mgr.h72 CollectiveExecutor* FindOrCreate(int64 step_id) override { in FindOrCreate()
86 void Cleanup(int64 step_id) override { in Cleanup()
123 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId()
Dscoped_allocator_mgr.cc129 void ScopedAllocatorMgr::Cleanup(int64 step_id) { in Cleanup()
138 ScopedAllocatorContainer* ScopedAllocatorMgr::GetContainer(int64 step_id) { in GetContainer()
153 const Tensor& backing_tensor, int64 step_id, int32 scope_id, in AddScopedAllocator()
Dsimple_propagator_state.cc26 const ImmutableExecutorState& immutable_state, int64 step_id, bool vlog) in SimplePropagatorState()
31 const ImmutableExecutorState& immutable_state, int64 step_id, in SimplePropagatorState()
Dcollective_rma_local.h30 int64 step_id) in CollectiveRemoteAccessLocal()
Dcollective_executor_mgr.h64 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId()
Dbase_collective_executor.h99 CollectiveRemoteAccess* remote_access, int64 step_id, in BaseCollectiveExecutor()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc140 const int64 step_id = 123; in TEST_F() local
165 const int64 step_id = 123; in TEST_F() local
179 const int64 step_id = 321; in TEST_F() local
199 const int64 step_id = 123; in TEST_F() local
223 const int64 step_id = 123; in TEST_F() local
259 const int64 step_id = 123; in TEST_F() local
291 const int64 step_id = 123; in TEST_F() local
310 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.cc43 RpcRemoteRendezvous(const WorkerEnv* env, int64 step_id) in RpcRemoteRendezvous()
62 void Init(WorkerInterface* wi, int64 step_id, StringPiece key, in Init()
313 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64 step_id, in Create()
Dgrpc_response_cache.h76 int64 step_id = -1; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtest_visitor_util.cc46 int step_id = 0; in runOnFunction() local
63 int step_id = 0; in runOnFunction() local
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto249 int64 step_id = 2; field
322 int64 step_id = 1; field
338 int64 step_id = 1; field
427 int64 step_id = 1; field
485 int64 step_id = 1; field

123