Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/python/eager/
Dtape_test.py174 var1.assign_add(1.0)
175 var2.assign_add(2.0)
183 var1.assign_add(1.0)
184 var2.assign_add(2.0)
192 var1.assign_add(1.0)
194 var2.assign_add(2.0)
205 var1.assign_add(1.0)
206 var2.assign_add(2.0)
Dremote_test.py258 v.assign_add(math_ops.reduce_mean(i))
281 v.assign_add(math_ops.reduce_mean(i))
307 v.assign_add(math_ops.reduce_mean(i))
498 variable_b.assign_add(i)
504 variable_b.assign_add(i)
590 v1.assign_add(1)
661 v1.assign_add(1)
677 v1.assign_add(1)
686 v2.assign_add(1)
704 v1.assign_add(1)
[all …]
Dwrap_function_test.py48 op = v.assign_add(x)
277 v.assign_add(1, name='increment', read_value=False)
289 v.assign_add(
320 v0.assign_add(1, name='increment_v0')
321 v1.assign_add(1, name='increment_v1')
467 return v.assign_add(x)
510 return v.assign_add(x)
/external/tensorflow/tensorflow/python/distribute/experimental/rpc/
Drpc_ops_test.py358 def assign_add(a): function
359 v.assign_add(a)
371 server.register("assign_add", assign_add)
381 result_or = client1_with_listed_methods.assign_add(
393 result_or = client2_with_listed_methods.assign_add(
500 def assign_add(a): function
501 v.assign_add(a)
510 server.register("assign_add", assign_add)
660 def assign_add(a): function
661 v.assign_add(a)
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dtpu_values.py165 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in TPUDistributedVariable
167 return self._primary.assign_add(value, use_locking, name, read_value)
168 return self._policy.assign_add(
263 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in TPUMirroredVariable
267 assign_add_fn = lambda v, *a, **ka: v.assign_add(*a, **ka)
284 return assign_add(
361 def assign_add(self, *args, **kwargs): member in TPUSyncOnReadVariable
363 return values.SyncOnReadVariable.assign_add(self, *args, **kwargs)
392 def assign_add(var, value, use_locking=False, name=None, read_value=True): function
439 def assign_add(self, member in TPUOnWritePolicy
[all …]
Dpacked_distributed_variable_test.py51 val0 = packed_var.assign(2.0).assign_add(1.0)
56 val0 = packed_var.assign(2.0).assign_add(1.0)
63 packed_var.assign_add(3.0).assign_sub(1.0)
97 var0.assign_add(3.0)
Dparameter_server_strategy_test.py150 x_add = x.assign_add(c)
166 y_add = y.assign_add(array_ops.identity(x_add))
180 z_add = z.assign_add(array_ops.identity(y))
246 x_add = x.assign_add(a, name='x_add')
305 x_add = x.assign_add(c)
319 y_add = y.assign_add(array_ops.identity(x_add))
334 z_add = z.assign_add(array_ops.identity(y))
396 x_add = x.assign_add(one, use_locking=True)
397 y_add = y.assign_add(one, use_locking=True)
398 z_add = z.assign_add(one, use_locking=True)
Ddistributed_variable_test.py196 self.assertIsInstance(v.assign_add(delta), core.Tensor)
255 v.assign_add(one, True, "assign", False)
256 v.assign_add(one, use_locking=True, name="assign", read_value=False)
287 assign_add_fn = lambda var, value: var.assign_add(value)
295 read_assign_fn = lambda var, value: var.assign_add(var.value() + var.
386 _test(lambda: check_ops.assert_equal_v2(v.assign_add(1.), 2.), v)
535 _assert_unsaveable(lambda: v.assign_add(ops.convert_to_tensor([1., 1.])))
Dpacked_distributed_variable.py156 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in PackedDistributedVariable
157 assign_add_fn = lambda var, *a, **kw: var.assign_add(*a, **kw)
310 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in PackedVarAndDevice
312 return self._var.assign_add(delta, use_locking, name, read_value)
/external/tensorflow/tensorflow/python/training/
Dmonitored_session_test.py292 do_step = state_ops.assign_add(gstep, 1)
307 new_gstep = state_ops.assign_add(gstep, 1)
324 new_gstep = state_ops.assign_add(gstep, 1)
359 w_add = w0.assign_add(1.)
361 new_gstep = state_ops.assign_add(gstep, 1)
382 new_gstep = state_ops.assign_add(gstep, 1)
400 new_gstep = state_ops.assign_add(gstep, 1)
422 do_step = state_ops.assign_add(gstep, 1)
443 new_gstep = state_ops.assign_add(gstep, 1)
458 new_gstep = state_ops.assign_add(gstep, 1)
[all …]
Devaluation_test.py183 eval_ops = state_ops.assign_add(my_var, 1.0)
211 eval_ops = [state_ops.assign_add(my_var, 1.0),
212 state_ops.assign_add(
Devaluation.py230 update_eval_step = state_ops.assign_add(
235 update_eval_step = state_ops.assign_add(eval_step, 1, use_locking=True)
/external/tensorflow/tensorflow/python/kernel_tests/variables/
Dvariable_ops_test.py155 added = state_ops.assign_add(var, [[2, 3]])
163 added = state_ops.assign_add(var, [[2.0, 3.0]])
171 added = state_ops.assign_add(var, self._NewShapelessTensor())
179 added = state_ops.assign_add(var, self._NewShapelessTensor())
191 var = state_ops.assign_add(var, [[6.0, 7.0]])
255 increment = state_ops.assign_add(var, [1.0])
Ddense_update_ops_no_tsan_test.py40 state_ops.assign_add(
106 state_ops.assign_add(
/external/tensorflow/tensorflow/python/grappler/
Dconstant_folding_test.py78 x.assign_add(gx)
79 y.assign_add(gy)
/external/tensorflow/tensorflow/python/debug/wrappers/
Ddisk_usage_test.py46 self.inc_v = state_ops.assign_add(self.v, self.delta, name="inc_v")
47 self.dec_v = state_ops.assign_add(self.v, self.eta, name="dec_v")
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py467 update_ops.append(variable.assign_add(tp))
470 update_ops.append(variable.assign_add(fp))
474 update_ops.append(variable.assign_add(tn))
478 update_ops.append(variable.assign_add(fn))
724 return var.assign_add(math_ops.reduce_sum(label_and_pred, 1))
/external/tensorflow/tensorflow/python/autograph/tests/
Dcond_basic_test.py29 v.assign_add(1)
36 v.assign_add(1)
38 v.assign_add(2)
/external/tensorflow/tensorflow/python/ops/
Dmetrics_impl.py273 update_op = state_ops.assign_add(total_cm, current_cm)
460 update_total_op = state_ops.assign_add(total, math_ops.reduce_sum(values))
462 update_count_op = state_ops.assign_add(count, num_values)
753 update_ops['tp'] = state_ops.assign_add(true_p,
765 update_ops['fn'] = state_ops.assign_add(false_n,
777 update_ops['tn'] = state_ops.assign_add(true_n,
789 update_ops['fp'] = state_ops.assign_add(false_p,
1579 update_total_op = state_ops.assign_add(total, values)
1581 update_count_op = state_ops.assign_add(count, num_values)
1691 update_op = state_ops.assign_add(count, math_ops.reduce_sum(values))
[all …]
Dstate_ops.py205 def assign_add(ref, value, use_locking=None, name=None): function
271 return gen_state_ops.assign_add(
273 return ref.assign_add(value)
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py193 self.updates = [state_ops.assign_add(self.iterations, 1)]
268 self.updates = [state_ops.assign_add(self.iterations, 1)]
344 self.updates = [state_ops.assign_add(self.iterations, 1)]
425 self.updates = [state_ops.assign_add(self.iterations, 1)]
533 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
629 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
712 with ops.control_dependencies([state_ops.assign_add(self.iterations, 1)]):
823 self.updates = [state_ops.assign_add(self.iterations, 1)]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmake_deterministic_test.py66 v.assign_add(1.)
102 v.assign_add(1.)
142 v.assign_add(1.)
199 v.assign_add(1.)
/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.py137 counter_var.assign_add(1)
163 counter_var.assign_add(1)
205 counter_var.assign_add(1)

123456