Searched refs:train_ops (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer_test.py | 38 train_ops = []
82 train_ops.append(train_op)
84 return sessions, graphs, train_ops
101 sessions, graphs, train_ops = get_workers(num_workers,
127 sessions[0].run(train_ops[0])
128 sessions[1].run(train_ops[1])
149 sessions[0].run(train_ops[0])
150 sessions[1].run(train_ops[1])
164 target=self._run, args=(train_ops[0], sessions[0])))
167 target=self._run, args=(train_ops[1], sessions[1])))
[all …]
|
| D | sync_replicas_optimizer.py | 254 train_ops = []
288 train_ops.append(grad_accum.apply_grad(
297 train_ops.append(grad_accum.apply_indexed_slices_grad(
323 with ops.control_dependencies(train_ops):
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | file_input_yielder.cc | 94 std::unordered_set<string> train_ops; in NextItem() local
97 train_ops.insert(NodeName(val)); in NextItem()
101 if (train_ops.find(node.name()) != train_ops.end()) { in NextItem()
105 if (train_ops_found.size() != train_ops.size()) { in NextItem()
106 for (const auto& train_op : train_ops) { in NextItem()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_test_lib.py | 426 train_ops, value = strategy.extended.call_for_each_replica(model_fn)
427 self.evaluate(strategy.group(train_ops))
|