Home
last modified time | relevance | path

Searched defs:step_id (Results 1 – 25 of 85) 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.cc44 void LogMemory::RecordStep(const int64_t step_id, const string& handle) { in RecordStep()
52 const int64_t step_id, in RecordTensorAllocation()
70 const int64_t step_id, const int index, in RecordTensorOutput()
81 const int64_t step_id, size_t num_bytes, in RecordRawAllocation()
94 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.cc154 [this](const int64_t step_id) { in InitItem()
356 Status GraphMgr::SendInputs(const int64_t step_id, const NamedTensors& in) { in SendInputs()
375 Status GraphMgr::RecvOutputs(const int64_t step_id, NamedTensors* out) { in RecvOutputs()
393 void GraphMgr::RecvOutputsAsync(const int64_t step_id, NamedTensors* out, in RecvOutputsAsync()
420 const string& handle, const int64_t step_id, const ExecutorOpts& opts, in ExecuteAsync()
517 const string& handle, int64_t step_id, Item* item, Rendezvous* rendezvous, in StartParallelExecutors()
Dbase_rendezvous_mgr.cc58 RemoteRendezvous* BaseRendezvousMgr::Find(int64_t step_id) { in Find()
62 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64_t step_id) { in FindOrCreate()
73 void BaseRendezvousMgr::RecvLocalAsync(int64_t step_id, in RecvLocalAsync()
87 Status BaseRendezvousMgr::RecvLocal(int64_t step_id, in RecvLocal()
106 void BaseRendezvousMgr::Cleanup(int64_t step_id) { in Cleanup()
129 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.cc111 void Worker::AbortStep(int64_t step_id) { in AbortStep()
169 const int64_t step_id = request->step_id(); in DoRunGraph() local
267 const int64_t step_id = request->step_id(); in DoPartialRunGraph() local
357 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.cc165 void RetrieveLogs(int64_t step_id, StepStats* ss) { in RetrieveLogs()
648 const FetchListType& fetches, const MasterEnv* env, int64_t step_id, in RunPartitionsHelper()
812 const MasterEnv* env, int64_t step_id, int64_t execution_count, in RunPartitions()
837 const MasterEnv* env, int64_t step_id, int64_t execution_count, in RunPartitions()
878 CleanupBroadcastHelper(int64_t step_id, int num_calls, StatusCallback done) in CleanupBroadcastHelper()
928 int64_t step_id, StatusCallback done) { in CleanupPartitionsAsync()
941 void MasterSession::ReffedClientGraph::ProcessStats(int64_t step_id, in ProcessStats()
1593 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId() local
1893 uint64 step_id, int64_t count, in FillPerStepState()
1920 uint64 step_id, in PostRunCleanup()
[all …]
/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.h84 CollectiveExecutor* FindOrCreate(int64_t step_id) override { in FindOrCreate()
98 void Cleanup(int64_t step_id) override { in Cleanup()
135 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/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.cc41 RpcRemoteRendezvous(const WorkerEnv* env, int64_t step_id) in RpcRemoteRendezvous()
60 void Init(WorkerInterface* wi, int64_t step_id, StringPiece key, in Init()
311 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64_t step_id, in Create()
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_compat_request_state.cc63 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()
104 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()
/external/tensorflow/tensorflow/core/profiler/lib/
Dscoped_memory_debug_annotation.h55 explicit ScopedMemoryDebugAnnotation(const char* op_name, int64_t step_id) { in ScopedMemoryDebugAnnotation()
81 const char* op_name, int64_t step_id, const char* region_type, in ScopedMemoryDebugAnnotation()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto251 int64 step_id = 2; field
324 int64 step_id = 1; field
340 int64 step_id = 1; field
429 int64 step_id = 1; field
487 int64 step_id = 1; field

1234