| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr.cc | 72 int64 graph_key, const StatusCallback& done) { in RefreshStepIdSequenceAsync() argument
76 auto it = sequence_table_.find(graph_key); in RefreshStepIdSequenceAsync()
78 gks = new GraphKeySequence(graph_key); in RefreshStepIdSequenceAsync()
79 sequence_table_[graph_key] = gks; in RefreshStepIdSequenceAsync()
89 req->add_graph_key(graph_key); in RefreshStepIdSequenceAsync()
114 for (int64 graph_key : request->graph_key()) { in GetStepSequenceAsync() local
115 auto it = sequence_table_.find(graph_key); in GetStepSequenceAsync()
118 gks = new GraphKeySequence(graph_key); in GetStepSequenceAsync()
120 sequence_table_[graph_key] = gks; in GetStepSequenceAsync()
125 ss->set_graph_key(graph_key); in GetStepSequenceAsync()
[all …]
|
| D | rpc_collective_executor_mgr.h | 52 void RefreshStepIdSequenceAsync(int64 graph_key,
55 int64 NextStepId(int64 graph_key) override;
57 void RetireStepId(int64 graph_key, int64 step_id) override;
|
| D | rpc_collective_executor_mgr_test.cc | 151 values[ss.graph_key()] = ss.next_step_id(); in TEST_F()
170 EXPECT_EQ(values[ss.graph_key()], ss.next_step_id()); in TEST_F()
|
| D | master_session.h | 142 uint64 NewStepId(int64 graph_key);
|
| D | master_session.cc | 1570 uint64 MasterSession::NewStepId(int64 graph_key) { in NewStepId() argument
1571 if (graph_key == BuildGraphOptions::kNoCollectiveGraphKey) { in NewStepId()
1575 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()
1581 graph_key, [&status, ¬e](const Status& s) { in NewStepId()
1593 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale.py | 163 graph_key = None
166 graph_key = graph._graph_key # pylint: disable=protected-access
168 key = (name, graph_key)
179 graph_key = None
182 graph_key = graph._graph_key # pylint: disable=protected-access
185 if g == graph_key:
195 graph_key = None
198 graph_key = graph._graph_key # pylint: disable=protected-access
199 return self._weights.get((name, graph_key), None)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | session_ops.py | 256 graph_key = TensorHandle._get_reader_key(handle)
257 result = graph._handle_readers.get(graph_key)
266 graph._handle_readers[graph_key] = result
279 graph_key = TensorHandle._get_mover_key(feeder, handle)
280 result = graph._handle_movers.get(graph_key)
287 graph._handle_movers[graph_key] = result
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.h | 57 void RefreshStepIdSequenceAsync(int64 graph_key,
60 int64 NextStepId(int64 graph_key) override { in NextStepId() argument
64 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() argument
|
| D | test_collective_executor_mgr.h | 114 void RefreshStepIdSequenceAsync(int64 graph_key, in RefreshStepIdSequenceAsync() argument
119 int64 NextStepId(int64 graph_key) override { in NextStepId() argument
123 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() argument
|
| D | collective_executor_mgr.cc | 90 int64 graph_key, const StatusCallback& done) { in RefreshStepIdSequenceAsync() argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 264 graph_key = None
267 graph_key = graph._graph_key # pylint: disable=protected-access
269 key = (name, graph_key)
279 graph_key = None
282 graph_key = graph._graph_key # pylint: disable=protected-access
285 if g == graph_key:
296 graph_key = None
299 graph_key = graph._graph_key # pylint: disable=protected-access
300 return self._weights.get((name, graph_key), None)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 219 virtual void RefreshStepIdSequenceAsync(int64 graph_key,
225 virtual int64 NextStepId(int64 graph_key) = 0;
230 virtual void RetireStepId(int64 graph_key, int64 step_id) = 0;
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 596 repeated int64 graph_key = 1; field
600 int64 graph_key = 1; field
|