Home
last modified time | relevance | path

Searched refs:assign_add (Results 1 – 25 of 141) sorted by relevance

123456

/external/tensorflow/tensorflow/python/autograph/converters/
Dside_effect_guards_test.py101 tf.assign_add(a, 1)
103 tf.assign_add(a, 1)
112 with self.compiled(node, {}, state_ops.assign_add) as result:
146 tf.assign_add(a, 1)
156 state_ops.assign_add) as result:
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_grpc_test.py390 inc_v_1 = state_ops.assign_add(v_1, delta_1, name="inc_v_1")
391 inc_v_2 = state_ops.assign_add(v_2, delta_2, name="inc_v_2")
463 inc_v_1 = state_ops.assign_add(v_1, toggled_1, name="inc_v_1")
464 inc_v_2 = state_ops.assign_add(v_2, toggled_2, name="inc_v_2")
498 inc_v = state_ops.assign_add(v, delta, name="inc_v")
543 inc_v_1 = state_ops.assign_add(v_1, delta_1, name="inc_v_1")
544 inc_v_2 = state_ops.assign_add(v_2, delta_2, name="inc_v_2")
601 inc_v_1 = state_ops.assign_add(v_1, delta_1, name="inc_v_1")
602 inc_v_2 = state_ops.assign_add(v_2, delta_2, name="inc_v_2")
674 inc_v_1 = state_ops.assign_add(v_1, delta_1, name="inc_v_1")
[all …]
/external/tensorflow/tensorflow/python/training/
Dmonitored_session_test.py278 do_step = state_ops.assign_add(gstep, 1)
293 new_gstep = state_ops.assign_add(gstep, 1)
310 new_gstep = state_ops.assign_add(gstep, 1)
329 new_gstep = state_ops.assign_add(gstep, 1)
347 new_gstep = state_ops.assign_add(gstep, 1)
369 do_step = state_ops.assign_add(gstep, 1)
418 new_gstep = state_ops.assign_add(gstep, 1)
439 new_gstep = state_ops.assign_add(gstep, 1)
459 new_gstep = state_ops.assign_add(gstep, 1)
479 new_gstep = state_ops.assign_add(gstep, 1)
[all …]
Devaluation_test.py189 eval_ops = state_ops.assign_add(my_var, 1.0)
217 eval_ops = [state_ops.assign_add(my_var, 1.0),
218 state_ops.assign_add(
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_ops_test.py152 added = state_ops.assign_add(var, [[2.0, 3.0]])
160 added = state_ops.assign_add(var, [[2.0, 3.0]])
168 added = state_ops.assign_add(var, self._NewShapelessTensor())
176 added = state_ops.assign_add(var, self._NewShapelessTensor())
187 var = state_ops.assign_add(var, [[6.0, 7.0]])
246 increment = state_ops.assign_add(var, [1.0])
Ddense_update_ops_no_tsan_test.py40 state_ops.assign_add(
100 state_ops.assign_add(
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py42 op = v.assign_add(x)
189 v.assign_add(1, name='increment', read_value=False)
201 v.assign_add(
232 v0.assign_add(1, name='increment_v0')
233 v1.assign_add(1, name='increment_v1')
335 return v.assign_add(x)
379 return v.assign_add(x)
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dgraph_actions_test.py508 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
527 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
540 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
554 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
567 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
583 state_ops.assign_add(variables_lib.get_global_step(), 1),
584 state_ops.assign_add(loss_var, -1.0))
603 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
629 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
655 train_op = state_ops.assign_add(global_step, 1)
Dmonitors_test.py266 tensor = state_ops.assign_add(var, 1.0)
531 assign_add = state_ops.assign_add(counter_var, 1.0, name='my_assign_add')
539 assign_add.name: step + 1.0,
549 assign_add.name: step + 31.0,
558 assign_add.name: (step + 31.0, step + 1.0),
564 assign_add.name: (step + 1.0, step + 31.0),
574 state_ops.assign_add(var, 1.0, name='my_assign_add')
624 self.train_op = state_ops.assign_add(self.global_step, 1)
788 inc_5 = state_ops.assign_add(global_step_tensor, 5)
/external/tensorflow/tensorflow/contrib/eager/python/
Dmetrics_impl.py256 self._vars[i].assign_add(math_ops.add_n([m._vars[i] for m in metrics]))
327 self.denom.assign_add(
330 self.numer.assign_add(math_ops.cast(values, self.dtype))
333 self.denom.assign_add(math_ops.reduce_sum(weights))
335 self.numer.assign_add(math_ops.reduce_sum(values))
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dautocast_variable_test.py165 self.assertAllClose(3.14 * 2, self.evaluate(x.assign_add(v1)))
176 self.evaluate(x.assign_add(v2))
184 self.assertAllClose(3.14 * 2, self.evaluate(x.assign_add(3.14)))
219 self.evaluate(x.assign_add(small_tensor)))
Dautocast_variable.py120 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in AutoCastVariable
121 return self._variable.assign_add(
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimators_test.py65 update_global_step = training_util.get_global_step().assign_add(1)
105 update_global_step = training_util.get_global_step().assign_add(1)
146 update_global_step = training_util.get_global_step().assign_add(1)
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dmoving_average_optimizer_test.py137 state_ops.assign_add(var0, [1.0, 1.0]),
138 state_ops.assign_add(var1, [2.0, 2.0]),
139 state_ops.assign_add(ema_var0, [3.0, 3.0]),
140 state_ops.assign_add(ema_var1, [4.0, 4.0])
Dagn_optimizer.py173 update_ops.append(state_ops.assign_add(grad_var, g))
193 update_ops.append(state_ops.assign_add(global_step, 1))
204 local_update = state_ops.assign_add(
/external/tensorflow/tensorflow/contrib/distribute/python/
Dparameter_server_strategy_test.py183 x_add = x.assign_add(c)
199 y_add = y.assign_add(array_ops.identity(x_add))
213 z_add = z.assign_add(array_ops.identity(y))
280 x_add = x.assign_add(a, name='x_add')
339 x_add = x.assign_add(c)
353 y_add = y.assign_add(array_ops.identity(x_add))
368 z_add = z.assign_add(array_ops.identity(y))
434 x_add = x.assign_add(one, use_locking=True)
435 y_add = y.assign_add(one, use_locking=True)
436 z_add = z.assign_add(one, use_locking=True)
/external/tensorflow/tensorflow/contrib/mixed_precision/python/
Dloss_scale_manager.py177 lambda: state_ops.assign_add(self._num_good_steps, 1).op)
192 state_ops.assign_add(self._num_bad_steps, 1),
/external/tensorflow/tensorflow/python/ops/
Dmetrics_impl.py276 update_op = state_ops.assign_add(total_cm, current_cm)
378 update_total_op = state_ops.assign_add(total, math_ops.reduce_sum(values))
380 update_count_op = state_ops.assign_add(count, num_values)
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,
1374 update_total_op = state_ops.assign_add(total, values)
1376 update_count_op = state_ops.assign_add(count, num_values)
1486 update_op = state_ops.assign_add(count, math_ops.reduce_sum(values))
[all …]
Dstate_ops.py166 def assign_add(ref, value, use_locking=None, name=None): function
189 return gen_state_ops.assign_add(
191 return ref.assign_add(value)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dmoving_stats.py98 mean_var = state_ops.assign_add(
173 return log_mean_exp_var.assign_add(x)
/external/tensorflow/tensorflow/python/debug/wrappers/
Ddisk_usage_test.py50 self.inc_v = state_ops.assign_add(self.v, self.delta, name="inc_v")
51 self.dec_v = state_ops.assign_add(self.v, self.eta, name="dec_v")
/external/tensorflow/tensorflow/python/keras/
Doptimizers.py187 self.updates = [state_ops.assign_add(self.iterations, 1)]
259 self.updates = [state_ops.assign_add(self.iterations, 1)]
327 self.updates = [state_ops.assign_add(self.iterations, 1)]
401 self.updates = [state_ops.assign_add(self.iterations, 1)]
488 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
579 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
659 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
749 self.updates = [state_ops.assign_add(self.iterations, 1)]
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc179 for (Node* assign_add : add_values_to_accumulator) { in rewriteNode()
180 g->AddControlEdge(assign_add, clean_up_accumulator); in rewriteNode()
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm.py42 sum_update = sum_metric.assign_add(scalar_tensor)
167 incr_step = state_ops.assign_add(training_util.get_global_step(), 1)
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops.py793 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,
3218 update_count = state_ops.assign_add(count_, batch_count) # n_AB in eqn
3227 update_mean_prediction = state_ops.assign_add(mean_prediction,
3237 update_mean_label = state_ops.assign_add(mean_label, delta_mean_label)
3257 update_comoment = state_ops.assign_add(comoment, delta_comoment)
3781 update_count_op = state_ops.assign_add(count_, num_values)
3890 update_po = state_ops.assign_add(po, po_t)
[all …]

123456