Searched refs:slot_variables (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/training/tracking/ |
D | graph_view.py | 92 slot_variables = object_identity.ObjectIdentityDictionary() 138 slot_variables.setdefault(trackable, []).append( 140 return slot_variables 342 slot_variables, argument 351 object_proto.slot_variables.extend(slot_variables.get(trackable, ())) 368 slot_variables = _serialize_slot_variables( 375 slot_variables=slot_variables) 453 slot_variables = _serialize_slot_variables( 457 return trackable_objects, path_to_root, node_ids, slot_variables 460 trackable_objects, _, node_ids, slot_variables = ( [all …]
|
D | util.py | 121 for slot_reference in node.slot_variables: 246 for slot_reference in node.slot_variables:
|
/external/tensorflow/tensorflow/python/tpu/ |
D | tpu_embedding.py | 2041 slot_variables = AdagradSlotVariable(accumulator_variables) 2090 return slot_variables, load_ops_fn, retrieve_ops_fn 2118 slot_variables = ProximalAdagradSlotVariable(accumulator_variables) 2167 return slot_variables, load_ops_fn, retrieve_ops_fn 2207 slot_variables = AdamSlotVariables(m_variables, v_variables) 2260 return slot_variables, load_ops_fn, retrieve_ops_fn 2307 slot_variables = FtrlSlotVariable(accumulator_variables, linear_variables) 2360 return slot_variables, load_ops_fn, retrieve_ops_fn 2403 slot_variables = ProximalYogiSlotVariables(v_variables, m_variables) 2456 return slot_variables, load_ops_fn, retrieve_ops_fn [all …]
|
/external/tensorflow/tensorflow/python/keras/tests/ |
D | tracking_util_with_v1_optimizers_test.py | 157 serialized_graph.nodes[optimizer_node.slot_variables[0] 163 serialized_graph.nodes[optimizer_node.slot_variables[0] 174 optimizer_node.slot_variables[0] 176 self.assertEqual("m", optimizer_node.slot_variables[0].slot_name) 180 optimizer_node.slot_variables[0]
|
D | tracking_util_test.py | 189 for slot in optimizer_node.slot_variables:
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | trackable_object_graph.proto | 56 repeated SlotVariableReference slot_variables = 3; field
|
D | saved_object_graph.proto | 53 slot_variables = 3; field
|
/external/tensorflow/tensorflow/python/saved_model/ |
D | load.py | 425 for slot_variable_proto in proto.slot_variables: 442 for slot_variable_proto in proto.slot_variables:
|
D | save.py | 285 object_proto.slot_variables.extend(self._slot_variables.get(node, ()))
|