Home
last modified time | relevance | path

Searched defs:assign_add (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/
H A Drpc_ops_test.py358 def assign_add(a): function
500 def assign_add(a): function
660 def assign_add(a): function
707 def assign_add(a): function
741 def assign_add(a): function
831 def assign_add(a): function
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_values.py165 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in TPUDistributedVariable
263 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in TPUMirroredVariable
361 def assign_add(self, *args, **kwargs): member in TPUSyncOnReadVariable
392 def assign_add(var, value, use_locking=False, name=None, read_value=True): function
439 def assign_add(self, member in TPUOnWritePolicy
549 def assign_add(self, var, *args, **kwargs): member in TPUOnReadPolicy
H A Dpacked_distributed_variable.py156 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in PackedDistributedVariable
310 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in PackedVarAndDevice
H A Dvalues.py781 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in DistributedVariable
1242 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in SyncOnReadVariable
1538 def assign_add(self, member in OnReadPolicy
1643 def assign_add(self, member in OnWritePolicy
H A Dps_values.py154 def assign_add(self, *args, **kwargs): member in AggregatingVariable
400 def assign_add(self, *args, **kwargs): member in CachingVariable
H A Dtpu_replicated_variable.py258 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in TPUReplicatedVariable
H A Dvalues_v2.py163 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in DistributedVariable
H A Dsharded_variable.py542 def assign_add(self, delta, use_locking=False, name=None, read_value=True): member in ShardedVariableMixin
H A Dcustom_training_loop_input_test.py111 def assign_add(): function
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dlite_v2_test_util.py106 def assign_add(self, x): member in ModelTest._getSimpleModelWithVariables.SimpleModelWithOneVariable
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_if_op_test.cc156 const auto assign_add = [](const string& fn_name, int v) { in TEST() local
H A Dlower_case_op_test.cc187 const auto assign_add = [](const string& fn_name, int v) { in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dvariables.py598 def assign_add(self, delta, use_locking=False, name=None, read_value=True): member in Variable
2056 def assign_add(self, delta, use_locking=False, name=None, read_value=True): member in RefVariable
3081 def assign_add(self, value, use_locking=False, name=None, read_value=True): member in PartitionedVariable
H A Dstate_ops.py205 def assign_add(ref, value, use_locking=None, name=None): function
H A Dresource_variable_ops.py873 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in BaseResourceVariable
2176 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in _UnreadVariable
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dautocast_variable.py251 def assign_add(self, delta, use_locking=None, name=None, read_value=True): member in AutoCastVariable
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction_test.py4833 def assign_add(): function