Home
last modified time | relevance | path

Searched refs:slot_variables (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/training/tracking/
Dgraph_view.py92 slot_variables = object_identity.ObjectIdentityDictionary()
135 slot_variables.setdefault(trackable, []).append(
137 return slot_variables
316 slot_variables, argument
325 object_proto.slot_variables.extend(slot_variables.get(trackable, ()))
341 slot_variables = _serialize_slot_variables(
348 slot_variables=slot_variables)
422 slot_variables = _serialize_slot_variables(
426 return trackable_objects, node_ids, slot_variables
Dutil_with_v1_optimizers_test.py163 serialized_graph.nodes[optimizer_node.slot_variables[0]
169 serialized_graph.nodes[optimizer_node.slot_variables[0]
180 optimizer_node.slot_variables[0]
182 self.assertEqual("m", optimizer_node.slot_variables[0].slot_name)
186 optimizer_node.slot_variables[0]
Dutil.py117 for slot_reference in node.slot_variables:
Dutil_test.py352 for slot in optimizer_node.slot_variables:
/external/tensorflow/tensorflow/contrib/checkpoint/python/
Dvisualize.py62 for slot_reference in node.slot_variables:
82 elif node.slot_variables:
90 for slot_reference in node.slot_variables:
/external/tensorflow/tensorflow/python/saved_model/
Dsave.py133 trackable_objects, node_ids, slot_variables = (
138 self.slot_variables = slot_variables
173 object_proto.slot_variables.extend(self.slot_variables.get(node, ()))
Dload.py139 for slot_variable_proto in proto.slot_variables:
156 for slot_variable_proto in proto.slot_variables:
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py825 slot_variables = AdagradSlotVariable(accumulator_variables)
868 return slot_variables, load_ops_fn, retrieve_ops_fn
913 slot_variables = AdamSlotVariables(m_variables, v_variables)
962 return slot_variables, load_ops_fn, retrieve_ops_fn
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dcheckpointable_utils_test.py181 serialized_graph.nodes[optimizer_node.slot_variables[0]
187 serialized_graph.nodes[optimizer_node.slot_variables[0]
198 optimizer_node.slot_variables[0]
200 self.assertEqual("m", optimizer_node.slot_variables[0].slot_name)
204 optimizer_node.slot_variables[0]
/external/tensorflow/tensorflow/core/protobuf/
Dtrackable_object_graph.proto55 repeated SlotVariableReference slot_variables = 3; field
Dsaved_object_graph.proto52 slot_variables = 3; field