def restore(self, restored_tensors, unused_restored_shapes):
# pylint: disable=protected-access
with ops.colocate_with(self.op._table_ref):
return gen_lookup_ops._lookup_table_import_v2(
self.op._table_ref, restored_tensors[0],
restored_tensors[1])
def restore(self, restore_tensors, shapes):
return gen_lookup_ops._lookup_table_import_v2(
self.op.table_ref, restore_tensors[0], restore_tensors[1])
def restore(self, restored_tensors, unused_restored_shapes):
# pylint: disable=protected-access
with ops.colocate_with(self.op._table_ref):
return gen_lookup_ops._lookup_table_import_v2(
self.op._table_ref, restored_tensors[0], restored_tensors[1])
def restore(self, restore_tensors, shapes):
return gen_lookup_ops._lookup_table_import_v2(
self.op.table_ref, restore_tensors[0], restore_tensors[1])
