Searched defs:var_device (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adam.py | 134 def _prepare_local(self, var_device, var_dtype, apply_state): argument
386 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | adamax.py | 113 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | ftrl.py | 139 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | adadelta.py | 100 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | nadam.py | 109 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | gradient_descent.py | 127 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | rmsprop.py | 163 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | adagrad.py | 86 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | optimizer_v2.py | 945 def _prepare_local(self, var_device, var_dtype, apply_state): argument
950 def _fallback_apply_state(self, var_device, var_dtype): argument
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2488 DeviceNameUtils::ParsedName var_device; in FindVariableInputs() local
|