Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/distribute/
Dtpu_values.py159 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in TPUDistributedVariable
161 return self._primary.assign_sub(value, use_locking, name, read_value)
162 return self._policy.assign_sub(
239 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in TPUMirroredVariable
243 assign_sub_fn = lambda v, *a, **ka: v.assign_sub(*a, **ka)
260 return assign_sub(
353 def assign_sub(self, *args, **kwargs): member in TPUSyncOnReadVariable
355 return values.SyncOnReadVariable.assign_sub(self, *args, **kwargs)
381 def assign_sub(var, value, use_locking=False, name=None, read_value=True): function
421 def assign_sub(self, member in TPUOnWritePolicy
[all …]
Dpacked_distributed_variable_test.py63 packed_var.assign_add(3.0).assign_sub(1.0)
66 packed_var.assign_sub(4.0).assign_sub(2.0)
Dpacked_distributed_variable.py147 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): member in PackedDistributedVariable
148 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw)
306 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): member in PackedVarAndDevice
308 return self._var.assign_sub(delta, use_locking, name, read_value)
Dmoving_averages_test.py201 w.assign_sub([0.5])
224 w.assign_sub([0.5])
241 w_assign = w.assign_sub([0.5])
Ddistributed_variable_test.py195 self.assertIsInstance(v.assign_sub(delta), core.Tensor)
257 v.assign_sub(one, True, "assign", False)
258 v.assign_sub(one, use_locking=True, name="assign", read_value=False)
291 assign_sub_fn = lambda var, value: var.assign_sub(value)
387 _test(lambda: check_ops.assert_equal_v2(v.assign_sub(1.), 1.), v)
536 _assert_unsaveable(lambda: v.assign_sub(ops.convert_to_tensor([1., 1.])))
Dtpu_replicated_variable.py240 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in TPUReplicatedVariable
242 assign_sub_fn = lambda var, *a, **ka: var.assign_sub(*a, **ka)
Dvalues_v2.py159 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): member in DistributedVariable
161 return super().assign_sub(delta, use_locking, name, read_value)
Dps_values.py150 def assign_sub(self, *args, **kwargs): member in AggregatingVariable
151 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw)
397 def assign_sub(self, *args, **kwargs): member in CachingVariable
398 return self._v.assign_sub(*args, **kwargs)
Dvalues.py768 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in DistributedVariable
770 return self._primary.assign_sub(value, use_locking, name, read_value)
772 return self._policy.assign_sub(
1229 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in SyncOnReadVariable
1231 return self._primary.assign_sub(value, use_locking, name, read_value)
1240 self).assign_sub(value, use_locking, name, read_value)
1517 def assign_sub(self, member in OnReadPolicy
1652 def assign_sub(self, member in OnWritePolicy
Dvalues_v2_test.py130 v.assign_sub(1., use_locking=True, name="assign_sub", read_value=True),
132 self.assertIsNone(v.assign_sub(1., read_value=False))
139 self.assertIsInstance(v.assign_sub(1., read_value=False), ops.Operation)
Dvalues_util.py157 assign_sub_fn = lambda var, *a, **kw: var.assign_sub(*a, **kw)
308 return variable.assign_sub(tensor)
Dtpu_replicated_variable_test.py64 self.evaluate(w.assign_sub(x1))
Dtpu_strategy_model_parallelism_test.py264 result = make_strategy_run(w.assign_sub)(delta)
295 w.assign_sub(x1)
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadam.py224 var_update = state_ops.assign_sub(
235 var_update = state_ops.assign_sub(
435 var.assign_sub(
457 var.assign_sub(coefficients['lr'] * m /
462 var.assign_sub(coefficients['lr'] * m /
/external/tensorflow/tensorflow/python/kernel_tests/variables/
Dvariable_ops_test.py157 subbed = state_ops.assign_sub(var, [[12, 13]])
165 subbed = state_ops.assign_sub(var, [[12.0, 13.0]])
173 subbed = state_ops.assign_sub(var, self._NewShapelessTensor())
181 subbed = state_ops.assign_sub(var, self._NewShapelessTensor())
Ddense_update_ops_test.py50 sub = state_ops.assign_sub(p, y)
/external/tensorflow/tensorflow/python/ops/
Dstate_ops.py133 def assign_sub(ref, value, use_locking=None, name=None): function
199 return gen_state_ops.assign_sub(
201 return ref.assign_sub(value)
Dstandard_ops.py84 from tensorflow.python.ops.state_ops import assign_sub
Dvariables.py616 def assign_sub(self, delta, use_locking=False, name=None, read_value=True): member in Variable
2078 def assign_sub(self, delta, use_locking=False, name=None, read_value=True): member in RefVariable
2094 assign = state_ops.assign_sub(
3089 def assign_sub(self, value, use_locking=False, name=None, read_value=True): member in PartitionedVariable
3090 assign_fn = lambda var, r_value: var.assign_sub(
/external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/
Ddense_update_ops_test.py56 a = state_ops.assign_sub(p, array_ops.fill([1024, 1024], 0))
/external/tensorflow/tensorflow/python/training/
Dmoving_averages.py91 return state_ops.assign_sub(v, (v - value) * decay, name=scope)
268 update_biased = state_ops.assign_sub(biased_var,
/external/tensorflow/tensorflow/python/eager/
Dremote_test.py591 v2.assign_sub(2)
662 v2.assign_sub(2)
705 v2.assign_sub(2)
744 v2.assign_sub(2)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.distribute.-sharded-variable.pbtxt41 name: "assign_sub"
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable.py255 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): member in AutoCastVariable
256 return self._apply_assign_update(self._variable.assign_sub, delta,
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dparallel_device_test.py600 param.assign_sub(0.01 * grad)
624 param.assign_sub(0.01 * grad)
657 param.assign_sub(0.01 * grad)

123