Home
last modified time | relevance | path

Searched refs:update_op (Results 1 – 23 of 23) sorted by relevance

/external/rust/crates/rayon/src/iter/
Dupdate.rs17 update_op: F, field
31 pub(super) fn new(base: I, update_op: F) -> Self { in new()
32 Update { base, update_op } in new()
47 let consumer1 = UpdateConsumer::new(consumer, &self.update_op); in drive_unindexed()
65 let consumer1 = UpdateConsumer::new(consumer, &self.update_op); in drive()
79 update_op: self.update_op, in with_producer()
84 update_op: F, in with_producer() field
100 update_op: &self.update_op, in with_producer()
112 update_op: &'f F, field
126 update_op: self.update_op, in into_iter()
[all …]
Dmod.rs774 fn update<F>(self, update_op: F) -> Update<Self, F> in update()
778 Update::new(self, update_op) in update()
/external/tensorflow/tensorflow/python/kernel_tests/
Dmetrics_test.py194 _, update_op = metrics.mean(
196 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
209 mean, update_op = metrics.mean(values)
213 self.evaluate(update_op)
227 mean, update_op = metrics.mean(values)
231 self.assertAlmostEqual(0.5, self.evaluate(update_op), 5)
232 self.assertAlmostEqual(1.475, self.evaluate(update_op), 5)
233 self.assertAlmostEqual(12.4 / 6.0, self.evaluate(update_op), 5)
234 self.assertAlmostEqual(1.65, self.evaluate(update_op), 5)
266 mean, update_op = mean_result
[all …]
/external/tensorflow/tensorflow/python/ops/
Dmetrics_impl.py276 update_op = state_ops.assign_add(total_cm, current_cm)
277 return total_cm, update_op
387 update_op = math_ops.div_no_nan(
391 ops.add_to_collections(updates_collections, update_op)
393 return mean_t, update_op
853 update_op = compute_auc(update_ops['tp'], update_ops['fn'],
857 ops.add_to_collections(updates_collections, update_op)
859 return auc_value, update_op
983 mean_distance, update_op = mean(radial_diffs, weights, None, None, name or
986 update_op = math_ops.subtract(1.0, update_op)
[all …]
/external/tensorflow/tensorflow/python/training/
Doptimizer.py108 def update_op(self, optimizer, g): member in _OptimizableVariable
125 def update_op(self, optimizer, g): member in _RefVariableProcessor
127 update_op = optimizer._apply_dense(g, self._v) # pylint: disable=protected-access
129 with ops.control_dependencies([update_op]):
132 return update_op
152 def update_op(self, optimizer, g): member in _DenseReadResourceVariableProcessor
154 update_op = optimizer._resource_apply_dense(g, self._v.op.inputs[0])
156 with ops.control_dependencies([update_op]):
159 return update_op
171 def update_op(self, optimizer, g): member in _DenseResourceVariableProcessor
[all …]
Devaluation_test.py120 accuracy, update_op = metrics_module.accuracy(labels, predictions)
126 eval_ops=update_op,
127 final_ops={'accuracy': (accuracy, update_op)},
152 accuracy, update_op = metrics_module.accuracy(labels, predictions)
158 eval_ops=update_op,
160 'accuracy': (accuracy, update_op),
Dsync_replicas_optimizer.py308 update_op = self._opt.apply_gradients(aggregated_grads_and_vars,
327 with ops.control_dependencies([update_op]):
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_output_test.py246 mean, update_op = metrics_module.mean_tensor(constant_op.constant([0]))
248 'metrics': (mean, update_op),
292 mean, update_op = metrics_module.mean_tensor(constant_op.constant([0]))
294 ('metrics', '1'): (mean, update_op),
315 mean, update_op = metrics_module.mean_tensor(constant_op.constant([0]))
317 'metrics_1': (mean, update_op),
335 mean, update_op = metrics_module.mean_tensor(constant_op.constant([0]))
337 'metrics_1': (mean, update_op),
374 mean, update_op = metrics_module.mean_tensor(constant_op.constant([0]))
376 'metrics_1': (mean, update_op),
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py68 update_op, should_apply = loss_scale.update([constant_op.constant(0.)])
69 self.evaluate(update_op)
76 update_op, should_apply = loss_scale.update(
78 self.evaluate(update_op)
139 update_op, should_apply_gradients = loss_scale.update([grad])
144 return array_ops.identity(update_op)
149 update_op = update()
155 self.evaluate(update_op)
Dloss_scale.py443 update_op = control_flow_ops.cond(is_finite, update_if_finite_grads,
446 return update_op, should_apply_gradients
/external/tensorflow/tensorflow/python/keras/
Dmetrics_confusion_matrix_test.py64 update_op = fp_obj.update_state(y_true, y_pred)
65 self.evaluate(update_op)
89 update_op = fp_obj.update_state(y_true, y_pred)
90 self.evaluate(update_op)
144 update_op = fn_obj.update_state(y_true, y_pred)
145 self.evaluate(update_op)
169 update_op = fn_obj.update_state(y_true, y_pred)
170 self.evaluate(update_op)
212 update_op = tn_obj.update_state(y_true, y_pred)
213 self.evaluate(update_op)
[all …]
Dmetrics_test.py73 update_op = m.update_state(ops.convert_to_tensor_v2_with_dispatch([1, 5]))
74 self.evaluate(update_op)
194 update_op = m.update_state([
198 self.evaluate(update_op)
378 update_op = acc_obj.update_state([[1], [2], [3], [4]], [[1], [2], [3], [4]])
379 self.evaluate(update_op)
402 update_op = acc_obj.update_state(rt1, rt2)
403 self.evaluate(update_op)
426 update_op = acc_obj.update_state([[1], [0]], [[1], [0]])
427 self.evaluate(update_op)
[all …]
Dmetrics.py217 update_op = None
219 update_op = self.update_state(*args, **kwargs) # pylint: disable=not-callable
221 if update_op is not None:
222 update_ops.append(update_op)
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py89 update_op = update_state_fn(*args, **kwargs)
90 if update_op is not None: # update_op will be None in eager execution.
91 metric_obj.add_update(update_op)
92 return update_op
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view_internal.h374 bool update_op = false; member
421 diff->update_op = false; in UpdateOp()
424 diff->update_op = true; in UpdateOp()
646 return !diff->update_name && !diff->update_op && !diff->update_device && in IsEmpty()
661 diff->update_op = false; in Reset()
Dgraph_view.cc833 diff.update_op ? diff.op : node->op(), device, in CheckKernelRegisteredForNodes()
1207 if (diff.update_op) { in ApplyNodeUpdates()
/external/tensorflow/tensorflow/python/framework/testdata/
Dmetrics_export_meta_graph.pb1483 name: "mean/update_op/e"
1514 name: "mean/update_op"
1518 input: "mean/update_op/e"
1557 key: "update_op"
1560 value: "mean/update_op:0"
/external/tensorflow/tensorflow/python/client/
Dsession_clusterspec_prop_test.py326 update_op = var.assign_add(feed)
340 sess1.run(update_op, feed_dict={feed: expected_ones}))
344 sess2.run(update_op, feed_dict={feed: expected_ones}))
346 sess1.run(update_op, feed_dict={feed: expected_ones}))
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py1650 update_op = math_ops.add
1652 update_op = math_ops.minimum
1654 update_op = math_ops.maximum
1660 update_op(tensor, updates),
1722 update_op = array_ops.tensor_scatter_update
1724 update_op = array_ops.tensor_scatter_add
1726 update_op = array_ops.tensor_scatter_min
1728 update_op = array_ops.tensor_scatter_max
1729 tensor = update_op(
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py696 update_op = self._resource_apply_dense(grad, var, **apply_kwargs)
698 with ops.control_dependencies([update_op]):
701 return update_op
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py571 update_op = state_ops.assign(v, x).op
572 return update_op
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py384 update_op = control_flow_ops.cond(is_finite, update_if_finite_grads,
387 return update_op, should_apply_gradients
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py951 _, update_op = metrics.mean(values)
955 self.evaluate(update_op)