Searched refs:_apply_sparse_shared (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | adam.py | 149 def _apply_sparse_shared(self, grad, var, indices, scatter_add, state): member in AdamOptimizer
176 return self._apply_sparse_shared(
188 return self._apply_sparse_shared(grad, var, indices,
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | adamax.py | 137 def _apply_sparse_shared(self, grad, var, indices, member in AdaMaxOptimizer
165 return self._apply_sparse_shared(
179 return self._apply_sparse_shared(
|
| D | adam_gs_optimizer.py | 179 def _apply_sparse_shared(self, grad, var, indices, scatter_add): member in AdamGSOptimizer
206 return self._apply_sparse_shared(
222 return self._apply_sparse_shared(grad, var, indices,
|
| D | shampoo.py | 163 return self._apply_sparse_shared(grad_values, grad_indices, var)
166 return self._apply_sparse_shared(grad.values, grad.indices, var)
168 def _apply_sparse_shared(self, grad_values, grad_indices, var): member in ShampooOptimizer
|
| D | nadam_optimizer.py | 71 def _apply_sparse_shared(self, grad, var, indices, scatter_add): member in NadamOptimizer
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam.py | 179 def _apply_sparse_shared(self, grad, var, indices, scatter_add): member in AdamOptimizer
206 return self._apply_sparse_shared(
222 return self._apply_sparse_shared(grad, var, indices,
|