Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_values.py189 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
331 def scatter_mul(self, *args, **kwargs): member in TPUMirroredVariable
512 def scatter_mul(self, var, sparse_delta, use_locking=False, name=None): member in TPUOnWritePolicy
579 def scatter_mul(self, *args, **kwargs): member in TPUOnReadPolicy
H A Dpacked_distributed_variable.py190 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
326 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
H A Dvalues.py825 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
1282 def scatter_mul(self, *args, **kwargs): member in SyncOnReadVariable
1582 def scatter_mul(self, *args, **kwargs): member in OnReadPolicy
1669 def scatter_mul(self, var, sparse_delta, use_locking=False, name=None): member in OnWritePolicy
H A Dvalues_v2.py179 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
H A Dvalues_util.py239 def scatter_mul(var, sparse_delta, use_locking=False, name=None): function
H A Dsharded_variable.py653 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in ShardedVariableMixin
H A Dvars_test.py556 def scatter_mul(): function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dstate_ops.py732 def scatter_mul(ref, indices, updates, use_locking=False, name=None): function
H A Dresource_variable_ops.py1058 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2206 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
H A Dvariables.py700 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in Variable
2198 def scatter_mul(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.py275 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable