| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| H A D | adam.py | 132 def _prepare_local(self, var_device, var_dtype, apply_state): argument
164 def _resource_apply_dense(self, grad, var, apply_state=None): argument
201 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
384 def _prepare_local(self, var_device, var_dtype, apply_state): argument
418 def _resource_apply_dense(self, grad, var, apply_state=None): argument
439 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | adamax.py | 110 def _prepare_local(self, var_device, var_dtype, apply_state): argument
130 def _resource_apply_dense(self, grad, var, apply_state=None): argument
149 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | ftrl.py | 153 def _prepare_local(self, var_device, var_dtype, apply_state): argument
167 def _resource_apply_dense(self, grad, var, apply_state=None): argument
205 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | adadelta.py | 89 def _prepare_local(self, var_device, var_dtype, apply_state): argument
106 def _resource_apply_dense(self, grad, var, apply_state=None): argument
123 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | nadam.py | 106 def _prepare_local(self, var_device, var_dtype, apply_state): argument
147 def _resource_apply_dense(self, grad, var, apply_state=None): argument
170 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | gradient_descent.py | 124 def _prepare_local(self, var_device, var_dtype, apply_state): argument
129 def _resource_apply_dense(self, grad, var, apply_state=None): argument
166 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | rmsprop.py | 160 def _prepare_local(self, var_device, var_dtype, apply_state): argument
173 def _resource_apply_dense(self, grad, var, apply_state=None): argument
219 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | adagrad.py | 91 def _prepare_local(self, var_device, var_dtype, apply_state): argument
132 def _resource_apply_dense(self, grad, var, apply_state=None): argument
146 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
|
| H A D | optimizer_v2.py | 682 def _distributed_apply(self, distribution, grads_and_vars, name, apply_state): argument
951 def _prepare_local(self, var_device, var_dtype, apply_state): argument
1240 def _resource_apply_dense(self, grad, handle, apply_state): argument
1282 def _resource_apply_sparse(self, grad, handle, indices, apply_state): argument
|