Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A 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
H A 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()
H A 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()
H A 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()
H A 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 …]
H A 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()
H A 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()
H A 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
H A Dmessage_wrappers.cc315 int64_t InMemoryRunGraphRequest::step_id() const { return step_id_; } in step_id() function in tensorflow::InMemoryRunGraphRequest
317 void InMemoryRunGraphRequest::set_step_id(int64_t step_id) { in set_step_id()
451 int64_t MutableProtoRunGraphRequest::step_id() const { in step_id() function in tensorflow::MutableProtoRunGraphRequest
455 void MutableProtoRunGraphRequest::set_step_id(int64_t step_id) { in set_step_id()
567 int64_t ProtoRunGraphRequest::step_id() const { return request_->step_id(); } in step_id() function in tensorflow::ProtoRunGraphRequest
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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
H A 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()
H A Drun_handler.cc739 int64_t step_id() const { return step_id_; } in step_id() function in tensorflow::RunHandler::Impl
834 int64_t step_id, int64_t timeout_in_ms, in Get()
1079 int64_t step_id, in Reset()
1097 int64_t step_id, int64_t timeout_in_ms, in Get()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A 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
H A 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()
H A 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()
H A Dgrpc_worker_service.cc447 const int64_t step_id = request->step_id(); in GrpcRecvTensorAsync() local
594 const int64_t step_id = request->step_id(); in RecvBufAsync() local
725 for (const auto& step_id : request->fetch_step_id()) { in LoggingAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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()
H A 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()
H A 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()
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler.cc664 int64_t step_id() const { return step_id_; } in step_id() function in tfrt::tf::RunHandler::Impl
771 std::unique_ptr<RunHandler> Get(int64_t step_id, int64_t timeout_in_ms, in Get()
1007 void RunHandler::Impl::Reset(int64_t step_id, in Reset()
1028 int64_t step_id, int64_t timeout_in_ms, const RunHandlerOptions& options) { in Get()
1053 int64_t RunHandler::step_id() const { return impl_->step_id(); } in step_id() function in tfrt::tf::RunHandler
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/lib/
H A 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()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dlog_memory.pb.h1132 inline ::int64_t MemoryLogStep::step_id() const { in step_id() function
1201 inline ::int64_t MemoryLogTensorAllocation::step_id() const { in step_id() function
1419 inline ::int64_t MemoryLogTensorOutput::step_id() const { in step_id() function
1588 inline ::int64_t MemoryLogRawAllocation::step_id() const { in step_id() function
1762 inline ::int64_t MemoryLogRawDeallocation::step_id() const { in step_id() function

1234