Home
last modified time | relevance | path

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

1234

/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_t step_id, const string& handle) { in RecordStep()
48 const int64_t step_id, in RecordTensorAllocation()
66 const int64_t step_id, const int index, in RecordTensorOutput()
77 const int64_t step_id, size_t num_bytes, in RecordRawAllocation()
90 const int64_t 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_t step_id, StepStats* ss) { in RetrieveLogs()
71 void WorkerCacheLogger::Save(const string& device, int64_t step_id, in Save()
85 void WorkerCacheLogger::RecordRecvTensor(int64_t step_id, int64_t start_usecs, in RecordRecvTensor()
95 void WorkerCacheLogger::RecordDataTransfer(int64_t step_id, int64_t start_usecs, in RecordDataTransfer()
Dgraph_mgr.cc152 [this](const int64_t step_id) { in InitItem()
352 Status GraphMgr::SendInputs(const int64_t step_id, const NamedTensors& in) { in SendInputs()
371 Status GraphMgr::RecvOutputs(const int64_t step_id, NamedTensors* out) { in RecvOutputs()
389 void GraphMgr::RecvOutputsAsync(const int64_t step_id, NamedTensors* out, in RecvOutputsAsync()
415 void GraphMgr::ExecuteAsync(const string& handle, const int64_t step_id, in ExecuteAsync()
512 const string& handle, int64_t step_id, Item* item, Rendezvous* rendezvous, in StartParallelExecutors()
Dbase_rendezvous_mgr.cc56 RemoteRendezvous* BaseRendezvousMgr::Find(int64_t step_id) { in Find()
60 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64_t step_id) { in FindOrCreate()
71 void BaseRendezvousMgr::RecvLocalAsync(int64_t step_id, in RecvLocalAsync()
85 Status BaseRendezvousMgr::RecvLocal(int64_t step_id, in RecvLocal()
104 void BaseRendezvousMgr::Cleanup(int64_t step_id) { in Cleanup()
120 int64_t step_id) in BaseRemoteRendezvous()
Drpc_collective_executor_mgr.cc50 CollectiveExecutor* RpcCollectiveExecutorMgr::Create(int64_t step_id) { in Create()
63 int64_t step_id = random::New64(); in NewRandomStepId() local
158 int64_t step_id) { in RetireStepId()
Dworker.cc109 void Worker::AbortStep(int64_t step_id) { in AbortStep()
165 const int64_t step_id = request->step_id(); in DoRunGraph() local
263 const int64_t step_id = request->step_id(); in DoPartialRunGraph() local
352 const int64_t step_id = request->step_id(); in CleanupGraphAsync() local
Dcollective_rma_distributed.h32 WorkerCacheInterface* worker_cache, int64_t step_id, string task_name) in CollectiveRemoteAccessDistributed()
Dworker_cache.h93 virtual bool RetrieveLogs(int64_t step_id, StepStats* ss) { return false; } in RetrieveLogs()
Dmaster_session.cc159 void RetrieveLogs(int64_t step_id, StepStats* ss) { in RetrieveLogs()
641 const FetchListType& fetches, const MasterEnv* env, int64_t step_id, in RunPartitionsHelper()
805 const MasterEnv* env, int64_t step_id, int64_t execution_count, in RunPartitions()
830 const MasterEnv* env, int64_t step_id, int64_t execution_count, in RunPartitions()
871 CleanupBroadcastHelper(int64_t step_id, int num_calls, StatusCallback done) in CleanupBroadcastHelper()
921 int64_t step_id, StatusCallback done) { in CleanupPartitionsAsync()
934 void MasterSession::ReffedClientGraph::ProcessStats(int64_t step_id, in ProcessStats()
1569 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId() local
1869 uint64 step_id, int64_t count, in FillPerStepState()
1896 uint64 step_id, in PostRunCleanup()
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc144 const int64_t step_id = 123; in TEST_F() local
169 const int64_t step_id = 123; in TEST_F() local
183 const int64_t step_id = 321; in TEST_F() local
203 const int64_t step_id = 123; in TEST_F() local
227 const int64_t step_id = 123; in TEST_F() local
263 const int64_t step_id = 123; in TEST_F() local
295 const int64_t step_id = 123; in TEST_F() local
314 const int64_t step_id = 123; in TEST_F() local
Dgrpc_response_cache.cc22 bool GrpcResponseCache::QueueRequest(int64_t request_id, int64_t step_id, in QueueRequest()
101 void GrpcResponseCache::CleanEntriesForStep(int64_t step_id) { in CleanEntriesForStep()
Drpc_rendezvous_mgr.cc43 RpcRemoteRendezvous(const WorkerEnv* env, int64_t step_id) in RpcRemoteRendezvous()
62 void Init(WorkerInterface* wi, int64_t step_id, StringPiece key, in Init()
313 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64_t step_id, in Create()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_executor_mgr.cc49 CollectiveExecutor* CollectiveExecutorMgr::FindOrCreate(int64_t step_id) { in FindOrCreate()
65 CollectiveExecutor* CollectiveExecutorMgr::Create(int64_t step_id) { in Create()
71 void CollectiveExecutorMgr::Cleanup(int64_t step_id) { in Cleanup()
Dtest_collective_executor_mgr.h80 CollectiveExecutor* FindOrCreate(int64_t step_id) override { in FindOrCreate()
94 void Cleanup(int64_t step_id) override { in Cleanup()
131 void RetireStepId(int64_t graph_key, int64_t step_id) override {} in RetireStepId()
Dscoped_allocator_mgr.cc130 void ScopedAllocatorMgr::Cleanup(int64_t step_id) { in Cleanup()
139 ScopedAllocatorContainer* ScopedAllocatorMgr::GetContainer(int64_t step_id) { in GetContainer()
154 const Tensor& backing_tensor, int64_t step_id, int32_t scope_id, in AddScopedAllocator()
Dsimple_propagator_state.cc26 const ImmutableExecutorState& immutable_state, int64_t step_id, bool vlog) in SimplePropagatorState()
31 const ImmutableExecutorState& immutable_state, int64_t step_id, in SimplePropagatorState()
Dcollective_rma_local.h30 int64_t step_id) in CollectiveRemoteAccessLocal()
Dcollective_executor_mgr.h63 void RetireStepId(int64_t graph_key, int64_t step_id) override {} in RetireStepId()
Dbase_collective_executor.h99 CollectiveRemoteAccess* remote_access, int64_t step_id, in BaseCollectiveExecutor()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtest_visitor_util.cc48 int step_id = 0; in runOnFunction() local
67 int step_id = 0; in runOnFunction() local
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_compat_request_state.cc67 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()
107 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()
/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

1234