Searched refs:_resource_scatter_add (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | nadam.py | 205 m_t = self._resource_scatter_add(m, indices, m_scaled_g_values)
218 v_t = self._resource_scatter_add(v, indices, v_scaled_g_values)
224 var_update = self._resource_scatter_add(
|
| D | rmsprop.py | 229 rms_t = self._resource_scatter_add(rms, indices, rms_scaled_g_values)
238 mg_t = self._resource_scatter_add(mg, indices, mg_scaled_g_values)
241 var_update = self._resource_scatter_add(
|
| D | adam.py | 240 m_t = self._resource_scatter_add(m, indices, m_scaled_g_values)
248 v_t = self._resource_scatter_add(v, indices, v_scaled_g_values)
|
| D | adamax.py | 178 var_update = self._resource_scatter_add(var, indices, var_slice)
|
| D | optimizer_v2.py | 979 def _resource_scatter_add(self, x, i, v): member in OptimizerV2
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam.py | 220 def _resource_scatter_add(self, x, i, v): member in AdamOptimizer
227 self._resource_scatter_add)
|