Home
last modified time | relevance | path

Searched defs:apply_state (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadam.py131 def _prepare_local(self, var_device, var_dtype, apply_state): argument
163 def _resource_apply_dense(self, grad, var, apply_state=None): argument
200 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
383 def _prepare_local(self, var_device, var_dtype, apply_state): argument
417 def _resource_apply_dense(self, grad, var, apply_state=None): argument
438 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Dadamax.py110 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
Dftrl.py153 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
Dadadelta.py89 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
Dnadam.py106 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
Dgradient_descent.py124 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
Drmsprop.py160 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
Dadagrad.py91 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
Doptimizer_v2.py681 def _distributed_apply(self, distribution, grads_and_vars, name, apply_state): argument
950 def _prepare_local(self, var_device, var_dtype, apply_state): argument
1241 def _resource_apply_dense(self, grad, handle, apply_state): argument
1283 def _resource_apply_sparse(self, grad, handle, indices, apply_state): argument