(self, table, name)
| 2105 | """SaveableObject implementation for DenseHashTable.""" |
| 2106 | |
| 2107 | def __init__(self, table, name): |
| 2108 | tensors = table.export() |
| 2109 | specs = [ |
| 2110 | BaseSaverBuilder.SaveSpec(tensors[0], "", name + "-keys"), |
| 2111 | BaseSaverBuilder.SaveSpec(tensors[1], "", name + "-values") |
| 2112 | ] |
| 2113 | # pylint: disable=protected-access |
| 2114 | super(DenseHashTable._Saveable, self).__init__(table, specs, name) |
| 2115 | |
| 2116 | def restore(self, restored_tensors, restored_shapes, name=None): |
| 2117 | del restored_shapes # unused |