Searched defs:var_device (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adam.py | 132 def _prepare_local(self, var_device, var_dtype, apply_state): argument
384 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | adamax.py | 110 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | ftrl.py | 153 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | adadelta.py | 89 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | nadam.py | 106 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | gradient_descent.py | 124 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | rmsprop.py | 160 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | adagrad.py | 91 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
| D | optimizer_v2.py | 951 def _prepare_local(self, var_device, var_dtype, apply_state): argument
956 def _fallback_apply_state(self, var_device, var_dtype): argument
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2779 DeviceNameUtils::ParsedName var_device; in FindVariableInputs() local
|