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.py134 def _prepare_local(self, var_device, var_dtype, apply_state): argument
166 def _resource_apply_dense(self, grad, var, apply_state=None): argument
203 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
386 def _prepare_local(self, var_device, var_dtype, apply_state): argument
420 def _resource_apply_dense(self, grad, var, apply_state=None): argument
441 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Dadamax.py113 def _prepare_local(self, var_device, var_dtype, apply_state): argument
133 def _resource_apply_dense(self, grad, var, apply_state=None): argument
152 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Dftrl.py139 def _prepare_local(self, var_device, var_dtype, apply_state): argument
153 def _resource_apply_dense(self, grad, var, apply_state=None): argument
191 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Dadadelta.py100 def _prepare_local(self, var_device, var_dtype, apply_state): argument
117 def _resource_apply_dense(self, grad, var, apply_state=None): argument
134 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Dnadam.py109 def _prepare_local(self, var_device, var_dtype, apply_state): argument
150 def _resource_apply_dense(self, grad, var, apply_state=None): argument
173 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Dgradient_descent.py127 def _prepare_local(self, var_device, var_dtype, apply_state): argument
132 def _resource_apply_dense(self, grad, var, apply_state=None): argument
169 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Drmsprop.py163 def _prepare_local(self, var_device, var_dtype, apply_state): argument
176 def _resource_apply_dense(self, grad, var, apply_state=None): argument
222 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Dadagrad.py86 def _prepare_local(self, var_device, var_dtype, apply_state): argument
127 def _resource_apply_dense(self, grad, var, apply_state=None): argument
141 def _resource_apply_sparse(self, grad, var, indices, apply_state=None): argument
Doptimizer_v2.py676 def _distributed_apply(self, distribution, grads_and_vars, name, apply_state): argument
945 def _prepare_local(self, var_device, var_dtype, apply_state): argument
1236 def _resource_apply_dense(self, grad, handle, apply_state): argument
1278 def _resource_apply_sparse(self, grad, handle, indices, apply_state): argument