Lines Matching refs:dataset
18 #include "tensorflow/core/kernels/data/dataset.h"
34 // Create a new DenseToSparseBatchDatasetOp::Dataset, insert it in the
61 *output = new Dataset<T>(ctx, batch_size, row_shape, input); \
79 class Dataset : public GraphDatasetBase {
81 Dataset(OpKernelContext* ctx, int64 batch_size,
95 ~Dataset() override { input_->Unref(); }
114 ")::Dataset");
137 class Iterator : public DatasetIterator<Dataset<T>> {
140 : DatasetIterator<Dataset<T>>(params),
141 input_impl_(params.dataset->input_->MakeIterator(params.prefix)) {}
151 DatasetIterator<Dataset<T>>::dataset()->batch_size_);
153 DatasetIterator<Dataset<T>>::dataset()->row_shape_;
172 i < DatasetIterator<Dataset<T>>::dataset()->batch_size_ &&
222 DatasetIterator<Dataset<T>>::dataset()->input_->output_dtypes()[0],
