| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 1078 def _slot_dict(self, slot_name): argument
1088 named_slots = self._slots.get(slot_name, None)
1091 self._slots[slot_name] = named_slots
1094 def _get_or_make_slot(self, var, val, slot_name, op_name): argument
1107 named_slots = self._slot_dict(slot_name)
1111 slot_name=slot_name, variable=var,
1117 slot_name, op_name): argument
1132 named_slots = self._slot_dict(slot_name)
1137 slot_name=slot_name, variable=var,
1142 def _zeros_slot(self, var, slot_name, op_name): argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 861 def add_slot(self, var, slot_name, initializer="zeros", shape=None): argument
877 if slot_name not in self._slot_names:
878 self._slot_names.append(slot_name)
881 weight = slot_dict.get(slot_name, None)
909 name="%s/%s" % (var._shared_name, slot_name), # pylint: disable=protected-access
914 slot_dict[slot_name] = weight
916 slot_name=slot_name, variable=var,
921 def get_slot(self, var, slot_name): argument
924 return slot_dict[slot_name]
1326 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | graph_view.py | 80 def _name_slot_variable(variable_path, slot_name): argument
84 + _escape_local_name(slot_name))
102 for slot_name in slot_names:
107 original_variable, slot_name)
127 slot_name=slot_name)
135 slot_name=slot_name,
|
| D | base.py | 282 slot_name=deferred_slot_restoration.slot_name)
295 slot_name=slot_restoration.slot_name))
307 slot_name=slot_restoration.slot_name)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 867 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
868 …return self._optimizer._restore_slot_variable(slot_name, variable, # pylint: disable=protected-ac…
871 def _create_or_restore_slot_variable(self, slot_variable_position, slot_name, argument
874 slot_variable_position, slot_name, variable)
876 def get_slot(self, var, slot_name): argument
877 return self._optimizer.get_slot(var, slot_name)
879 def add_slot(self, var, slot_name, initializer='zeros'): argument
880 return self._optimizer.add_slot(var, slot_name, initializer)
1147 def _create_or_restore_slot_variable(self, slot_variable_position, slot_name, argument
1150 slot_variable_position, slot_name, variable)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.optimizers.-adam.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-adamax.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-s-g-d.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-adagrad.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-adadelta.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-r-m-sprop.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-optimizer.pbtxt | 32 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
56 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.optimizers.-adagrad.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-r-m-sprop.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-s-g-d.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.optimizers.-adam.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-nadam.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.optimizers.-adadelta.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.optimizers.-s-g-d.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-adam.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.optimizers.-r-m-sprop.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.optimizers.-nadam.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.optimizers.-adamax.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-optimizer.pbtxt | 32 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
56 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.keras.optimizers.-ftrl.pbtxt | 33 …argspec: "args=[\'self\', \'var\', \'slot_name\', \'initializer\', \'shape\'], varargs=None, keywo…
57 argspec: "args=[\'self\', \'var\', \'slot_name\'], varargs=None, keywords=None, defaults=None"
|