Searched defs:scatter_add (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/python/training/ |
D | adam.py | 183 def _apply_sparse_shared(self, grad, var, indices, scatter_add): argument
|
/external/tensorflow/tensorflow/python/ops/ |
D | state_ops.py | 372 def scatter_add(ref, indices, updates, use_locking=False, name=None): function
|
D | resource_variable_ops.py | 876 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable 1896 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
|
D | variables.py | 644 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in Variable 2131 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in RefVariable
|
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
D | autocast_variable.py | 203 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
|