示例#1
0
 def _restore_op(self, iterator_resource):
   iterator_state_variant = parsing_ops.parse_tensor(
       io_ops.read_file(self._iterator_checkpoint_prefix_local()),
       dtypes.variant)
   restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
                                                     iterator_state_variant)
   return restore_op
示例#2
0
 def restore(self, restored_tensors, unused_restored_shapes):
   with ops.colocate_with(self.op):
     return gen_dataset_ops.deserialize_iterator(self.op, restored_tensors[0])
示例#3
0
 def _restore_op(iterator_resource):
   iterator_state_variant = parsing_ops.parse_tensor(
       io_ops.read_file(_path()), dtypes.variant)
   restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
                                                     iterator_state_variant)
   return restore_op
示例#4
0
 def restore(self, restored_tensors, restored_shapes):
     with ops.colocate_with(self.op):
         return gen_dataset_ops.deserialize_iterator(
             self.op, restored_tensors[0])
示例#5
0
 def _restore_op(iterator_resource):
   iterator_state_variant = parsing_ops.parse_tensor(
       io_ops.read_file(_path()), dtypes.variant)
   restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
                                                     iterator_state_variant)
   return restore_op