def __init__(self, iterator_resource, name): serialized_iterator = gen_dataset_ops.serialize_iterator( iterator_resource) specs = [ BaseSaverBuilder.SaveSpec(serialized_iterator, "", name + "_STATE") ] super(_IteratorSaveable, self).__init__(iterator_resource, specs, name)
def _save_op(self, iterator_resource): iterator_state_variant = gen_dataset_ops.serialize_iterator( iterator_resource) save_op = io_ops.write_file( self._iterator_checkpoint_prefix_local(), parsing_ops.serialize_tensor(iterator_state_variant)) return save_op
def __init__(self, iterator_resource, name): serialized_iterator = gen_dataset_ops.serialize_iterator(iterator_resource) specs = [ BaseSaverBuilder.SaveSpec(serialized_iterator, "", name + "_STATE") ] # pylint: disable=protected-access super(_IteratorSaveable, self).__init__(iterator_resource, specs, name)
def __init__(self, iterator_resource): serialized_iterator = gen_dataset_ops.serialize_iterator(iterator_resource) specs = [ saver.BaseSaverBuilder.SaveSpec(serialized_iterator, "", iterator_resource.name + "-state") ] super(_Saveable, self).__init__(iterator_resource, specs, iterator_resource.name)
def __init__(self, iterator_resource): serialized_iterator = gen_dataset_ops.serialize_iterator(iterator_resource) specs = [ saver_lib.BaseSaverBuilder.SaveSpec(serialized_iterator, "", iterator_resource.name + "-state") ] super(_Saveable, self).__init__(iterator_resource, specs, iterator_resource.name)
def __init__(self, iterator_resource, name): serialized_iterator = gen_dataset_ops.serialize_iterator( iterator_resource) specs = [ BaseSaverBuilder.SaveSpec(serialized_iterator, "", name + "_STATE") ] # pylint: disable=protected-access super(Iterator._Saveable, self).__init__(iterator_resource, specs, name)
def __init__(self, iterator_resource, name, external_state_policy=options_lib.ExternalStatePolicy.FAIL): serialized_iterator = gen_dataset_ops.serialize_iterator( iterator_resource, external_state_policy=external_state_policy.value) specs = [ BaseSaverBuilder.SaveSpec(serialized_iterator, "", name + "_STATE", device=iterator_resource.device) ] super(_IteratorSaveable, self).__init__(iterator_resource, specs, name)
def _save_op(iterator_resource): iterator_state_variant = gen_dataset_ops.serialize_iterator( iterator_resource) save_op = io_ops.write_file( _path(), parsing_ops.serialize_tensor(iterator_state_variant)) return save_op
def _save_op(iterator_resource): iterator_state_variant = gen_dataset_ops.serialize_iterator( iterator_resource) save_op = io_ops.write_file( _path(), parsing_ops.serialize_tensor(iterator_state_variant)) return save_op