using System;

using System.Collections.Generic;

using System.Text;

using static Tensorflow.Binding;

namespace Tensorflow

{

/// <summary>

/// Creates a `Dataset` that prefetches elements from this dataset.

/// </summary>

public class PrefetchDataset : UnaryUnchangedStructureDataset

{

Tensor _buffer_size;

public PrefetchDataset(IDatasetV2 input_dataset,

long buffer_size = -1,

int? slack_period = null) :

base(input_dataset)

{

_buffer_size = tf.convert_to_tensor(buffer_size, dtype: TF_DataType.TF_INT64, name: "buffer_size");

variant_tensor = ops.prefetch_dataset(input_dataset.variant_tensor,

_buffer_size,

input_dataset.output_types,

input_dataset.output_shapes,

slack_period: slack_period);

}

}

}