Searched defs:batch_scatter_update (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | state_ops.py | 948 def batch_scatter_update(ref, indices, updates, use_locking=True, name=None): function
|
H A D | resource_variable_ops.py | 1130 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable 2221 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
|
H A D | variables.py | 748 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): member in Variable 2270 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): member in RefVariable
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
H A D | values_v2.py | 199 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
|
H A D | sharded_variable.py | 683 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): member in ShardedVariableMixin
|
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/ |
H A D | autocast_variable.py | 287 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
|