Searched refs:_resource_scatter_add (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adagrad.py | 156 def _resource_scatter_add(x, i, v): function
166 acc_t = _resource_scatter_add(acc, indices, math_ops.square(grad))
168 var_update = _resource_scatter_add(
|
| D | nadam.py | 176 m_t = self._resource_scatter_add(m, indices, m_scaled_g_values)
187 v_t = self._resource_scatter_add(v, indices, v_scaled_g_values)
193 var_update = self._resource_scatter_add(
|
| D | rmsprop.py | 217 rms_t = self._resource_scatter_add(rms, indices, rms_scaled_g_values)
225 mg_t = self._resource_scatter_add(mg, indices, mg_scaled_g_values)
228 var_update = self._resource_scatter_add(
|
| D | adam.py | 220 m_t = self._resource_scatter_add(m, indices, m_scaled_g_values)
227 v_t = self._resource_scatter_add(v, indices, v_scaled_g_values)
|
| D | adamax.py | 163 var_update = self._resource_scatter_add(var, indices, var_slice)
|
| D | optimizer_v2.py | 835 def _resource_scatter_add(self, x, i, v): member in OptimizerV2
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | weight_decay_optimizers.py | 203 def _resource_scatter_add(self, x, i, v, _=None): member in DecoupledWeightDecayExtension
211 scatter_add = self._resource_scatter_add
|
| D | adam_gs_optimizer.py | 216 def _resource_scatter_add(self, x, i, v): member in AdamGSOptimizer
223 self._resource_scatter_add)
|
| D | adamax.py | 181 self._resource_scatter_add, self._resource_scatter_update)
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | adam.py | 182 def _resource_scatter_add(self, x, i, v): member in AdamOptimizer
189 self._resource_scatter_add, state)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam.py | 216 def _resource_scatter_add(self, x, i, v): member in AdamOptimizer
223 self._resource_scatter_add)
|