Searched refs:slot_variable (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | graph_view.py | 106 slot_variable = trackable.get_slot(
109 slot_variable = None
110 if slot_variable is None:
112 slot_variable._maybe_initialize_trackable() # pylint: disable=protected-access
113 if slot_variable._checkpoint_dependencies: # pylint: disable=protected-access
119 if slot_variable in node_ids:
124 "you.") % slot_variable)
128 object_names[slot_variable] = checkpoint_name
130 node_ids[slot_variable] = slot_variable_node_id
131 trackable_objects.append(slot_variable)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 1112 slot_variable=new_slot_variable)
1138 slot_variable=new_slot_variable)
1160 slot_variable=new_slot_variable)
1168 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
1178 checkpoint_position.restore(slot_variable)
1203 slot_variable = named_slots.get(variable_key, None)
1204 if (slot_variable is None and context.executing_eagerly() and
1219 slot_variable = self._get_or_make_slot_with_initializer(
1233 if slot_variable is not None:
1236 slot_variable_position.restore(slot_variable)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 917 slot_variable=weight)
1326 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
1336 checkpoint_position.restore(slot_variable)
1361 slot_variable = slot_dict.get(slot_name, None)
1362 if (slot_variable is None and context.executing_eagerly() and
1382 slot_variable = self.add_slot(
1393 if slot_variable is not None:
1396 slot_variable_position.restore(slot_variable)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 867 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
869 slot_variable)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 445 slot_variable = optimizer_object.add_slot(
448 nodes[slot_variable_proto.slot_variable_node_id] = slot_variable
|