Home
last modified time | relevance | path

Searched refs:graph_key (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Drpc_collective_executor_mgr.cc69 int64 graph_key, const StatusCallback& done) { in RefreshStepIdSequenceAsync() argument
73 auto it = sequence_table_.find(graph_key); in RefreshStepIdSequenceAsync()
75 gks = new GraphKeySequence(graph_key); in RefreshStepIdSequenceAsync()
76 sequence_table_[graph_key] = gks; in RefreshStepIdSequenceAsync()
86 req->add_graph_key(graph_key); in RefreshStepIdSequenceAsync()
111 for (int64 graph_key : request->graph_key()) { in GetStepSequenceAsync() local
112 auto it = sequence_table_.find(graph_key); in GetStepSequenceAsync()
115 gks = new GraphKeySequence(graph_key); in GetStepSequenceAsync()
117 sequence_table_[graph_key] = gks; in GetStepSequenceAsync()
122 ss->set_graph_key(graph_key); in GetStepSequenceAsync()
[all …]
Drpc_collective_executor_mgr.h51 void RefreshStepIdSequenceAsync(int64 graph_key,
54 int64 NextStepId(int64 graph_key) override;
56 void RetireStepId(int64 graph_key, int64 step_id) override;
Drpc_collective_executor_mgr_test.cc148 values[ss.graph_key()] = ss.next_step_id(); in TEST_F()
167 EXPECT_EQ(values[ss.graph_key()], ss.next_step_id()); in TEST_F()
Dmaster_session.h144 uint64 NewStepId(int64 graph_key);
Dmaster_session.cc1478 uint64 MasterSession::NewStepId(int64 graph_key) { in NewStepId() argument
1479 if (graph_key == BuildGraphOptions::kNoCollectiveGraphKey) { in NewStepId()
1483 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()
1489 graph_key, [&status, &note](const Status& s) { in NewStepId()
1501 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()
/external/tensorflow/tensorflow/python/ops/
Dsession_ops.py257 graph_key = TensorHandle._get_reader_key(handle)
258 result = graph._handle_readers.get(graph_key)
267 graph._handle_readers[graph_key] = result
280 graph_key = TensorHandle._get_mover_key(feeder, handle)
281 result = graph._handle_movers.get(graph_key)
288 graph._handle_movers[graph_key] = result
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_executor_mgr.h49 void RefreshStepIdSequenceAsync(int64 graph_key,
52 int64 NextStepId(int64 graph_key) override { in NextStepId() argument
56 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() argument
Dtest_collective_executor_mgr.h101 void RefreshStepIdSequenceAsync(int64 graph_key, in RefreshStepIdSequenceAsync() argument
106 int64 NextStepId(int64 graph_key) override { in NextStepId() argument
110 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() argument
Dcollective_executor_mgr.cc86 int64 graph_key, const StatusCallback& done) { in RefreshStepIdSequenceAsync() argument
/external/tensorflow/tensorflow/core/framework/
Dcollective.h194 virtual void RefreshStepIdSequenceAsync(int64 graph_key,
200 virtual int64 NextStepId(int64 graph_key) = 0;
205 virtual void RetireStepId(int64 graph_key, int64 step_id) = 0;
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Doptimizer_v2.py901 graph_key = graph._graph_key # pylint: disable=protected-access
904 if graph_key in self._per_graph_state:
905 per_graph_state = self._per_graph_state[graph_key]
909 self._per_graph_state[graph_key] = per_graph_state
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto545 repeated int64 graph_key = 1; field
549 int64 graph_key = 1; field