Home
last modified time | relevance | path

Searched refs:assign_sub (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Dtpu_values.py194 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in TPUDistributedVariable
196 return self._primary.assign_sub(value, use_locking, name, read_value)
197 return self._policy.assign_sub(
258 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in TPUMirroredVariable
268 return assign_sub(
338 def assign_sub(self, *args, **kwargs): member in TPUSyncOnReadVariable
340 return values.SyncOnReadVariable.assign_sub(self, *args, **kwargs)
366 def assign_sub(var, value, use_locking=False, name=None, read_value=True): function
405 def assign_sub(self, member in TPUOnWritePolicy
420 return assign_sub(
[all …]
Dpacked_distributed_variable_test.py66 packed_var.assign_add(3.0).assign_sub(1.0)
69 packed_var.assign_sub(4.0).assign_sub(2.0)
Dpacked_distributed_variable.py151 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): member in PackedDistributedVariable
152 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw)
294 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): member in PackedVarAndDevice
296 return self._var.assign_sub(delta, use_locking, name, read_value)
Dmoving_averages_test.py210 w.assign_sub([0.5])
233 w.assign_sub([0.5])
250 w_assign = w.assign_sub([0.5])
Dvalues.py730 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in DistributedVariable
732 return self._primary.assign_sub(value, use_locking, name, read_value)
734 return self._policy.assign_sub(
1153 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in SyncOnReadVariable
1155 return self._primary.assign_sub(value, use_locking, name, read_value)
1164 self).assign_sub(value, use_locking, name, read_value)
1424 def assign_sub(self, var, value, use_locking=False, name=None, member in OnReadPolicy
1539 def assign_sub(self, var, value, use_locking=False, name=None, member in OnWritePolicy
Dps_values.py131 def assign_sub(self, *args, **kwargs): member in AggregatingVariable
132 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw)
Dvalues_util.py161 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw)
312 return variable.assign_sub(tensor)
Dsharded_variable.py531 def assign_sub(self, delta, use_locking=False, name=None, read_value=True): member in ShardedVariableMixin
533 v.assign_sub(
Dvalues_test.py518 self.assertIsInstance(v.assign_sub(delta), core.Tensor)
581 v.assign_sub(one, True, "assign", False)
582 v.assign_sub(one, use_locking=True, name="assign", read_value=False)
615 assign_sub_fn = lambda var, value: var.assign_sub(value)
713 _test(lambda: check_ops.assert_equal_v2(v.assign_sub(1.), 1.), v)
860 _assert_unsaveable(lambda: v.assign_sub(ops.convert_to_tensor([1., 1.])))
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable_test.py187 self.assertEqual(evaluate(x.assign_sub(3)), 7)
284 self.assertAllClose(3., self.evaluate(x.assign_sub(v1)))
298 self.evaluate(x.assign_sub(v2))
304 self.assertAllClose(3., self.evaluate(x.assign_sub(3.)))
317 assign_sub = x.assign_sub(3.)
318 self.assertAllClose(3. * 2, self.evaluate(assign_sub))
320 self.evaluate(x.assign_sub(3.).assign_sub(3.)))
327 self.assertIsNone(self.evaluate(x.assign_sub(3., read_value=False)))
335 self.assertAllClose(3., self.evaluate(state_ops.assign_sub(x, 3.)))
356 return x.assign(1.).assign_add(3.).assign_add(3.).assign_sub(2.)
[all …]
Dautocast_variable.py260 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): member in AutoCastVariable
261 return self._apply_assign_update(self._variable.assign_sub, delta,
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadam.py226 var_update = state_ops.assign_sub(
237 var_update = state_ops.assign_sub(
437 var.assign_sub(
459 var.assign_sub(coefficients['lr'] * m /
464 var.assign_sub(coefficients['lr'] * m /
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_ops_test.py160 subbed = state_ops.assign_sub(var, [[12.0, 13.0]])
168 subbed = state_ops.assign_sub(var, [[12.0, 13.0]])
176 subbed = state_ops.assign_sub(var, self._NewShapelessTensor())
184 subbed = state_ops.assign_sub(var, self._NewShapelessTensor())
Ddense_update_ops_test.py56 sub = state_ops.assign_sub(p, y)
115 a = state_ops.assign_sub(p, array_ops.fill([1024, 1024], 0))
Dvariables_test.py137 minus_one = var.assign_sub(2.0)
156 minus_one = var.assign_sub(2.0)
828 minus_delta = pv_0.assign_sub(deltas_b)
835 assign_part_var = pv_1.assign_sub(pv_0)
Dresource_variable_ops_test.py969 self.evaluate(v.assign_sub(1.0))
973 assign_with_read = v.assign_sub(1.0, read_value=True)
975 assign_without_read = v.assign_sub(1.0, read_value=False)
1042 self.assertEqual(self.evaluate(var.assign_sub(1.).assign_sub(1.)), 3)
1209 new_v2_val = v2.assign_sub(v.read_value())
/external/tensorflow/tensorflow/python/ops/
Dstate_ops.py137 def assign_sub(ref, value, use_locking=None, name=None): function
162 return gen_state_ops.assign_sub(
164 return ref.assign_sub(value)
Dvariables.py612 def assign_sub(self, delta, use_locking=False, name=None, read_value=True): member in Variable
2084 def assign_sub(self, delta, use_locking=False, name=None, read_value=True): member in RefVariable
2100 assign = state_ops.assign_sub(
3095 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in PartitionedVariable
3096 assign_fn = lambda var, r_value: var.assign_sub(
Dstandard_ops.py89 from tensorflow.python.ops.state_ops import assign_sub
/external/tensorflow/tensorflow/python/eager/
Dremote_test.py551 v2.assign_sub(2)
622 v2.assign_sub(2)
648 v2.assign_sub(2)
687 v2.assign_sub(2)
/external/tensorflow/tensorflow/python/training/
Dmoving_averages.py94 return state_ops.assign_sub(v, (v - value) * decay, name=scope)
271 update_biased = state_ops.assign_sub(biased_var,
Dadam.py208 var_update = state_ops.assign_sub(
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dparallel_device_test.py468 param.assign_sub(0.01 * grad)
492 param.assign_sub(0.01 * grad)
526 param.assign_sub(0.01 * grad)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-variable.pbtxt72 name: "assign_sub"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-variable.pbtxt71 name: "assign_sub"

12