| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | training.py | 372 update_ops=None, argument
415 if update_ops is None:
416 update_ops = global_update_ops
418 update_ops = set(update_ops)
419 if not global_update_ops.issubset(update_ops):
424 if update_ops:
425 with ops.control_dependencies(update_ops):
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | agn_optimizer.py | 169 update_ops = []
170 update_ops.append(local_update_op)
173 update_ops.append(state_ops.assign_add(grad_var, g))
190 update_ops = []
193 update_ops.append(state_ops.assign_add(global_step, 1))
197 update_ops.append(state_ops.assign(lvar, g_val))
199 update_ops.append(
201 variable_update = control_flow_ops.group(*(update_ops))
210 with ops.control_dependencies(update_ops):
|
| D | model_average_optimizer.py | 177 update_ops = []
181 update_ops.append(state_ops.assign(lvar, gvar.read_value()))
182 return control_flow_ops.group(*(update_ops))
247 update_ops = []
251 update_ops.append(state_ops.assign(gvar, avg_var))
253 update_ops.append(state_ops.assign_add(global_step, 1))
254 with ops.control_dependencies(update_ops), ops.device(
|
| D | elastic_average_optimizer.py | 320 update_ops = []
328 update_ops.append(
334 update_ops.append(
340 update_ops.append(state_ops.assign_add(global_step, 1))
341 variable_update = control_flow_ops.group(*(update_ops))
|
| D | adamax.py | 183 def _finish(self, update_ops, name_scope): argument
185 with ops.control_dependencies(update_ops):
190 return control_flow_ops.group(*update_ops + [update_beta1],
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 377 update_ops = []
381 update_ops.append(result)
382 return update_ops
545 update_ops = [self._hashtable.insert(example_ids_hashed, esu)]
552 update_ops += self._get_partitioned_update_ops(
556 update_ops.append(state_ops.scatter_add(w, i, u))
563 update_ops.append(state_ops.assign_add(v, split_update))
565 update_ops.append(state_ops.assign_add(w, u))
567 return control_flow_ops.group(*update_ops)
568 with ops.control_dependencies(update_ops):
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | metrics_impl.py | 571 update_ops = {}
580 update_ops['tp'] = state_ops.assign_add(true_p,
592 update_ops['fn'] = state_ops.assign_add(false_n,
604 update_ops['tn'] = state_ops.assign_add(true_n,
616 update_ops['fp'] = state_ops.assign_add(false_p,
621 return values, update_ops
722 values, update_ops = _confusion_matrix_at_thresholds(
825 update_op = compute_auc(update_ops['tp'], update_ops['fn'],
826 update_ops['tn'], update_ops['fp'], 'update_op')
1590 values, update_ops = _confusion_matrix_at_thresholds(
[all …]
|
| D | accumulate_n_benchmark.py | 45 update_ops = [
49 with ops.control_dependencies(update_ops):
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| D | metric_ops.py | 784 update_ops = {}
793 update_ops['tp'] = state_ops.assign_add(true_positives,
805 update_ops['fn'] = state_ops.assign_add(false_negatives,
817 update_ops['tn'] = state_ops.assign_add(true_negatives,
829 update_ops['fp'] = state_ops.assign_add(false_positives,
834 return values, update_ops
841 values, update_ops = _streaming_confusion_matrix_at_thresholds(
843 return values['tp'], update_ops['tp']
850 values, update_ops = _streaming_confusion_matrix_at_thresholds(
852 return values['fn'], update_ops['fn']
[all …]
|
| /external/tensorflow/tensorflow/contrib/metrics/python/metrics/ |
| D | classification.py | 156 …values, update_ops = metrics_impl._confusion_matrix_at_thresholds( # pylint: disable=protected-ac…
180 update_op = compute_best_f1_score(tp=update_ops['tp'], fp=update_ops['fp'],
181 fn=update_ops['fn'], name='update')
|
| /external/tensorflow/tensorflow/python/training/ |
| D | evaluation.py | 64 def _get_latest_eval_step_value(update_ops): argument
74 if isinstance(update_ops, dict):
75 update_ops = list(update_ops.values())
77 with ops.control_dependencies(update_ops):
|
| D | optimizer.py | 578 update_ops = []
594 update_ops.append(processor.update_op(self, grad))
596 apply_updates = self._finish(update_ops, name)
598 with ops.control_dependencies([self._finish(update_ops, "update")]):
684 update_ops = [
691 def finish(self, update_ops): argument
692 return self._finish(update_ops, "update")
696 non_slot_devices, finish, args=(self, update_ops), group=False)
1036 def _finish(self, update_ops, name_scope): argument
1051 return control_flow_ops.group(*update_ops, name=name_scope)
|
| D | adam.py | 225 def _finish(self, update_ops, name_scope): argument
227 with ops.control_dependencies(update_ops):
235 *update_ops + [update_beta1, update_beta2], name=name_scope)
|
| /external/tensorflow/tensorflow/contrib/constrained_optimization/python/ |
| D | external_regret_optimizer.py | 262 update_ops = []
276 update_ops.append(
298 update_ops.append(
300 update_ops.append(
304 with ops.control_dependencies(update_ops):
|
| D | swap_regret_optimizer.py | 377 update_ops = []
391 update_ops.append(
413 update_ops.append(
415 update_ops.append(
419 with ops.control_dependencies(update_ops):
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | optimizers.py | 65 update_ops=None, argument
164 if update_ops is None:
165 update_ops = set(ops.get_collection(ops.GraphKeys.UPDATE_OPS))
167 if update_ops:
168 loss = control_flow_ops.with_dependencies(list(update_ops), loss)
|
| D | rev_block_lib_test.py | 359 update_ops = ops.get_collection_ref(ops.GraphKeys.UPDATE_OPS)
360 update_ops.pop()
361 update_ops.pop()
371 update_ops = ops.get_collection(ops.GraphKeys.UPDATE_OPS)
372 self.assertEqual(2, len(update_ops))
|
| D | optimizers_test.py | 322 update_ops=[update_op])
350 update_ops=[update_op])
368 update_ops=[update_op])
391 update_ops=[update_op])
414 update_ops=[update_op],
439 update_ops=[update_op])
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 333 update_ops = []
361 update_ops.append(
364 return control_flow_ops.group(update_ops)
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train.py | 904 update_ops = set(kwargs['update_ops'])
907 update_ops = set(ops.get_collection(ops.GraphKeys.UPDATE_OPS))
913 unused_ops = update_ops - all_gen_ops - all_dis_ops
917 gen_update_ops = list(all_gen_ops & update_ops)
918 dis_update_ops = list(all_dis_ops & update_ops)
1013 update_ops=gen_update_ops,
1035 update_ops=dis_update_ops,
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | optimizer_v2.py | 974 update_ops = []
976 update_ops.extend(distribution.extended.update(
987 update_ops = control_flow_ops.group(update_ops)
988 with ops.control_dependencies([update_ops]):
994 finish_updates = (update_ops,)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1146 update_ops = [are_all_splits_ready]
1152 update_ops.append(update_model)
1158 update_ops.append(self._finalized_trees.assign(stats.num_trees))
1159 update_ops.append(self._attempted_trees.assign(stats.attempted_trees))
1160 update_ops.append(training_state.num_layers.assign(stats.num_layers))
1161 update_ops.append(training_state.active_tree.assign(stats.active_tree))
1162 update_ops.append(
1166 update_ops.extend(
1169 return control_flow_ops.group(*update_ops, name="update_ensemble")
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | learning.py | 377 update_ops=None, argument
432 update_ops=update_ops,
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 437 update_ops = []
443 update_ops.extend(
448 tf_utils.is_symbolic_tensor(i) for i in update_ops)
453 … with ops._get_graph_from_inputs(update_ops).as_default(): # pylint: disable=protected-access
454 with ops.control_dependencies(update_ops):
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator.py | 856 update_ops = []
862 update_ops.append(metric_ops[1])
871 if update_ops:
872 update_ops = control_flow_ops.group(*update_ops)
874 update_ops = None
876 return update_ops, value_ops
|