Searched refs:_resource_scatter_add (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | nadam.py | 189 m_t = self._resource_scatter_add(m, indices, m_scaled_g_values)
202 v_t = self._resource_scatter_add(v, indices, v_scaled_g_values)
208 var_update = self._resource_scatter_add(
|
| D | rmsprop.py | 261 rms_t = self._resource_scatter_add(rms, indices, rms_scaled_g_values)
270 mg_t = self._resource_scatter_add(mg, indices, mg_scaled_g_values)
273 var_update = self._resource_scatter_add(
|
| D | adam.py | 214 m_t = self._resource_scatter_add(m, indices, m_scaled_g_values)
222 v_t = self._resource_scatter_add(v, indices, v_scaled_g_values)
|
| D | adamax.py | 176 var_update = self._resource_scatter_add(var, indices, var_slice)
|
| D | optimizer_v2.py | 1299 def _resource_scatter_add(self, x, i, v): member in OptimizerV2
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam.py | 223 def _resource_scatter_add(self, x, i, v): member in AdamOptimizer
230 self._resource_scatter_add)
|