Home
last modified time | relevance | path

Searched defs:graph_key (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Drpc_collective_executor_mgr.cc71 int64_t graph_key, const StatusCallback& done) { in RefreshStepIdSequenceAsync()
113 for (int64_t graph_key : request->graph_key()) { in GetStepSequenceAsync() local
148 int64_t RpcCollectiveExecutorMgr::NextStepId(int64_t graph_key) { in NextStepId()
157 void RpcCollectiveExecutorMgr::RetireStepId(int64_t graph_key, in RetireStepId()
H A Dmaster_session.cc1588 uint64 MasterSession::NewStepId(int64_t graph_key) { in NewStepId()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dtest_collective_executor_mgr.h126 void RefreshStepIdSequenceAsync(int64_t graph_key, in RefreshStepIdSequenceAsync()
131 int64_t NextStepId(int64_t graph_key) override { in NextStepId()
135 void RetireStepId(int64_t graph_key, int64_t step_id) override {} in RetireStepId()
H A Dcollective_executor_mgr.h59 int64_t NextStepId(int64_t graph_key) override { in NextStepId()
63 void RetireStepId(int64_t graph_key, int64_t step_id) override {} in RetireStepId()
H A Dcollective_executor_mgr.cc92 int64_t graph_key, const StatusCallback& done) { in RefreshStepIdSequenceAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dworker.proto598 repeated int64 graph_key = 1; field
602 int64 graph_key = 1; field
/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/protobuf/
Dworker.pb.h10750 inline ::int64_t GetStepSequenceRequest::graph_key(int index) const { in graph_key() function
10770 GetStepSequenceRequest::graph_key() const { in graph_key() function
10795 inline ::int64_t StepSequence::graph_key() const { in graph_key() function