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.py132 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
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.py682 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