Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadamax.py115 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Dftrl.py148 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Dadadelta.py111 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Dadam.py161 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Dnadam.py124 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Dgradient_descent.py103 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Drmsprop.py132 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Dadagrad.py107 def _prepare_local(self, var_device, var_dtype, apply_state): argument
Doptimizer_v2.py629 def _prepare_local(self, var_device, var_dtype, apply_state): argument
634 def _fallback_apply_state(self, var_device, var_dtype): argument