D | saver_test_utils.py | 7 # http://www.apache.org/licenses/LICENSE-2.0 35 # pylint: disable=protected-access 59 def insert(self, keys, values): argument 60 return gen_lookup_ops.lookup_table_insert_v2(self.table_ref, keys, values) 62 def lookup(self, keys, default): argument 63 return gen_lookup_ops.lookup_table_find_v2(self.table_ref, keys, default) 65 def keys(self): member in CheckpointedOp 82 name + "-keys"), 84 name + "-values") 88 def restore(self, restore_tensors, shapes): member in CheckpointedOp.CustomSaveable [all …]
|