Searched refs:_apply_update (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 242 def _apply_update(self, update_fn, *args, **kwargs): member in AutoCastVariable
265 return self._apply_update(self._variable.scatter_sub, sparse_delta,
269 return self._apply_update(self._variable.scatter_add, sparse_delta,
273 return self._apply_update(self._variable.scatter_max, sparse_delta,
277 return self._apply_update(self._variable.scatter_min, sparse_delta,
281 return self._apply_update(self._variable.scatter_mul, sparse_delta,
285 return self._apply_update(self._variable.scatter_div, sparse_delta,
289 return self._apply_update(self._variable.scatter_update, sparse_delta,
293 return self._apply_update(self._variable.batch_scatter_update, sparse_delta,
297 return self._apply_update(self._variable.scatter_nd_sub, indices, updates,
[all …]
|