Home
last modified time | relevance | path

Searched defs:scatter_update (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_values.py213 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
341 def scatter_update(self, *args, **kwargs): member in TPUMirroredVariable
522 def scatter_update(self, var, sparse_delta, use_locking=False, name=None): member in TPUOnWritePolicy
585 def scatter_update(self, *args, **kwargs): member in TPUOnReadPolicy
H A Dvalues.py861 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
1120 def scatter_update(self, *args, **kwargs): member in MirroredVariable
1302 def scatter_update(self, *args, **kwargs): member in SyncOnReadVariable
1598 def scatter_update(self, *args, **kwargs): member in OnReadPolicy
1695 def scatter_update(self, var, sparse_delta, use_locking=False, name=None): member in OnWritePolicy
H A Dpacked_distributed_variable.py222 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
342 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
H A Dvalues_v2.py195 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
H A Dvalues_util.py275 def scatter_update(var, sparse_delta, use_locking=False, name=None): function
H A Dsharded_variable.py673 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in ShardedVariableMixin
H A Dvars_test.py637 def scatter_update(v): function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dstate_ops.py383 def scatter_update(ref, indices, updates, use_locking=True, name=None): function
H A Dresource_variable_ops.py1106 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2216 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
H A Dvariables.py732 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in Variable
2246 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in RefVariable
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dautocast_variable.py283 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable