Searched refs:slot_variable (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | graph_view.py | 104 slot_variable = trackable.get_slot(
107 slot_variable = None
108 if slot_variable is None:
110 slot_variable._maybe_initialize_trackable() # pylint: disable=protected-access
111 if slot_variable._checkpoint_dependencies: # pylint: disable=protected-access
117 if slot_variable in node_ids:
125 object_names[slot_variable] = checkpoint_name
127 node_ids[slot_variable] = slot_variable_node_id
128 trackable_objects.append(slot_variable)
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | optimizer_v2.py | 293 slot_name=slot_name, variable=var, slot_variable=new_slot_variable)
324 slot_name=slot_name, variable=var, slot_variable=new_slot_variable)
346 slot_name=slot_name, variable=var, slot_variable=new_slot_variable)
376 slot_variable = self.get_slot(var=variable, name=slot_name)
377 if (slot_variable is None and context.executing_eagerly() and
390 slot_variable = self.create_slot(
398 if slot_variable is not None:
401 slot_variable_position.restore(slot_variable)
461 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
471 checkpoint_position.restore(slot_variable)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 1091 slot_variable=new_slot_variable)
1117 slot_variable=new_slot_variable)
1138 slot_variable=new_slot_variable)
1146 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
1156 checkpoint_position.restore(slot_variable)
1181 slot_variable = named_slots.get(variable_key, None)
1182 if (slot_variable is None and context.executing_eagerly() and
1195 slot_variable = self._get_or_make_slot(
1207 if slot_variable is not None:
1210 slot_variable_position.restore(slot_variable)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 552 slot_variable=weight)
849 def _restore_slot_variable(self, slot_name, variable, slot_variable): argument
859 checkpoint_position.restore(slot_variable)
884 slot_variable = slot_dict.get(slot_name, None)
885 if (slot_variable is None and context.executing_eagerly() and
898 slot_variable = self.add_slot(
909 if slot_variable is not None:
912 slot_variable_position.restore(slot_variable)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 159 slot_variable = optimizer_object.add_slot(
162 nodes[slot_variable_proto.slot_variable_node_id] = slot_variable
|