Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/python/eager/
Dtape_test.py177 var1.assign_add(1.0)
178 var2.assign_add(2.0)
186 var1.assign_add(1.0)
187 var2.assign_add(2.0)
195 var1.assign_add(1.0)
197 var2.assign_add(2.0)
208 var1.assign_add(1.0)
209 var2.assign_add(2.0)
Dremote_test.py218 v.assign_add(math_ops.reduce_mean(i))
241 v.assign_add(math_ops.reduce_mean(i))
267 v.assign_add(math_ops.reduce_mean(i))
458 variable_b.assign_add(i)
464 variable_b.assign_add(i)
550 v1.assign_add(1)
621 v1.assign_add(1)
647 v1.assign_add(1)
686 v1.assign_add(1)
Dwrap_function_test.py52 op = v.assign_add(x)
281 v.assign_add(1, name='increment', read_value=False)
293 v.assign_add(
324 v0.assign_add(1, name='increment_v0')
325 v1.assign_add(1, name='increment_v1')
471 return v.assign_add(x)
514 return v.assign_add(x)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable_test.py186 self.assertEqual(evaluate(x.assign_add(2)), 10)
283 self.assertAllClose(3. * 2, self.evaluate(x.assign_add(v1)))
294 self.evaluate(x.assign_add(v2))
303 self.assertAllClose(3. * 2, self.evaluate(x.assign_add(3.)))
312 assign_add = x.assign_add(3.)
313 self.assertAllClose(3., self.evaluate(assign_add))
315 self.evaluate(x.assign_add(3.).assign_add(3.)))
325 self.assertIsNone(self.evaluate(x.assign_add(2., read_value=False)))
334 self.evaluate(state_ops.assign_add(x, 3.)))
356 return x.assign(1.).assign_add(3.).assign_add(3.).assign_sub(2.)
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dtpu_values.py200 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in TPUDistributedVariable
202 return self._primary.assign_add(value, use_locking, name, read_value)
203 return self._policy.assign_add(
271 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in TPUMirroredVariable
281 return assign_add(
346 def assign_add(self, *args, **kwargs): member in TPUSyncOnReadVariable
348 return values.SyncOnReadVariable.assign_add(self, *args, **kwargs)
377 def assign_add(var, value, use_locking=False, name=None, read_value=True): function
423 def assign_add(self, member in TPUOnWritePolicy
438 return assign_add(
[all …]
Dpacked_distributed_variable_test.py54 val0 = packed_var.assign(2.0).assign_add(1.0)
59 val0 = packed_var.assign(2.0).assign_add(1.0)
66 packed_var.assign_add(3.0).assign_sub(1.0)
100 var0.assign_add(3.0)
Dparameter_server_strategy_test.py154 x_add = x.assign_add(c)
170 y_add = y.assign_add(array_ops.identity(x_add))
184 z_add = z.assign_add(array_ops.identity(y))
250 x_add = x.assign_add(a, name='x_add')
309 x_add = x.assign_add(c)
323 y_add = y.assign_add(array_ops.identity(x_add))
338 z_add = z.assign_add(array_ops.identity(y))
400 x_add = x.assign_add(one, use_locking=True)
401 y_add = y.assign_add(one, use_locking=True)
402 z_add = z.assign_add(one, use_locking=True)
Dpacked_distributed_variable.py160 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in PackedDistributedVariable
161 assign_add_fn = lambda var, *a, **kw: var.assign_add(*a, **kw)
298 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in PackedVarAndDevice
300 return self._var.assign_add(delta, use_locking, name, read_value)
Dmirrored_variable_test.py324 update0 = v0.assign_add(5.0 * replica_id_plus_one())
325 update1 = v1.assign_add(7.0 * replica_id_plus_one())
355 update0 = state_ops.assign_add(v0, 11.0 * replica_id_plus_one())
356 update1 = state_ops.assign_add(v1, 13.0 * replica_id_plus_one())
460 v_sum.assign_add(2.0 + replica_id),
/external/tensorflow/tensorflow/python/training/
Dmonitored_session_test.py294 do_step = state_ops.assign_add(gstep, 1)
309 new_gstep = state_ops.assign_add(gstep, 1)
326 new_gstep = state_ops.assign_add(gstep, 1)
345 new_gstep = state_ops.assign_add(gstep, 1)
363 new_gstep = state_ops.assign_add(gstep, 1)
385 do_step = state_ops.assign_add(gstep, 1)
406 new_gstep = state_ops.assign_add(gstep, 1)
421 new_gstep = state_ops.assign_add(gstep, 1)
464 new_gstep = state_ops.assign_add(gstep, 1)
485 new_gstep = state_ops.assign_add(gstep, 1)
[all …]
Devaluation_test.py187 eval_ops = state_ops.assign_add(my_var, 1.0)
215 eval_ops = [state_ops.assign_add(my_var, 1.0),
216 state_ops.assign_add(
Devaluation.py234 update_eval_step = state_ops.assign_add(
239 update_eval_step = state_ops.assign_add(eval_step, 1, use_locking=True)
/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/kernel_tests/
Dvariable_ops_test.py158 added = state_ops.assign_add(var, [[2.0, 3.0]])
166 added = state_ops.assign_add(var, [[2.0, 3.0]])
174 added = state_ops.assign_add(var, self._NewShapelessTensor())
182 added = state_ops.assign_add(var, self._NewShapelessTensor())
193 var = state_ops.assign_add(var, [[6.0, 7.0]])
252 increment = state_ops.assign_add(var, [1.0])
Ddense_update_ops_no_tsan_test.py44 state_ops.assign_add(
110 state_ops.assign_add(
Ddense_update_ops_test.py47 add = state_ops.assign_add(p, y)
107 a = state_ops.assign_add(p, array_ops.fill([1024, 1024], 0))
Dcritical_section_test.py53 nv = v.assign_add(a * b)
82 nv = v.assign_add(a * b)
145 nv = v.assign_add(a * b)
/external/tensorflow/tensorflow/python/grappler/
Dconstant_folding_test.py82 x.assign_add(gx)
83 y.assign_add(gy)
/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/
Doptimizer_v1.py198 self.updates = [state_ops.assign_add(self.iterations, 1)]
269 self.updates = [state_ops.assign_add(self.iterations, 1)]
344 self.updates = [state_ops.assign_add(self.iterations, 1)]
423 self.updates = [state_ops.assign_add(self.iterations, 1)]
521 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
615 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
697 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
807 self.updates = [state_ops.assign_add(self.iterations, 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,
1402 update_total_op = state_ops.assign_add(total, values)
1404 update_count_op = state_ops.assign_add(count, num_values)
1514 update_op = state_ops.assign_add(count, math_ops.reduce_sum(values))
[all …]
Dstate_ops.py168 def assign_add(ref, value, use_locking=None, name=None): function
193 return gen_state_ops.assign_add(
195 return ref.assign_add(value)
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op_test.cc156 const auto assign_add = [](const string& fn_name, int v) { in TEST() local
174 *(f_lib_proto.add_function()) = assign_add("AddOne", 1); in TEST()
175 *(f_lib_proto.add_function()) = assign_add("AddTwo", 2); in TEST()
Dlower_case_op_test.cc187 const auto assign_add = [](const string& fn_name, int v) { in TEST() local
205 *(f_lib_proto.add_function()) = assign_add("AddOne", 1); in TEST()
206 *(f_lib_proto.add_function()) = assign_add("AddTwo", 2); in TEST()
207 *(f_lib_proto.add_function()) = assign_add("AddTen", 10); in TEST()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dreduce_test.py141 counter_var.assign_add(1)
167 counter_var.assign_add(1)
209 counter_var.assign_add(1)

123456