Home | History | Annotate | Download | only in data

Lines Matching refs:dataset

19 #include "tensorflow/core/kernels/data/dataset.h"
52                                         "in the input dataset's elements (",
72                     "dataset's elements (",
82                       " and input dataset's component ", i, ": ",
88     *output = new Dataset(ctx, batch_size, std::move(padded_shapes),
93   class Dataset : public GraphDatasetBase {
95     Dataset(OpKernelContext* ctx, int64 batch_size,
106       // semantics. If we could tell statically that the input dataset
120     ~Dataset() override { input_->Unref(); }
138                              ")::Dataset");
186     class Iterator : public DatasetIterator<Dataset> {
189           : DatasetIterator<Dataset>(params),
190             input_impl_(params.dataset->input_->MakeIterator(params.prefix)) {}
205             batch_elements.reserve(dataset()->batch_size_);
206             for (int i = 0; i < dataset()->batch_size_ && !*end_of_sequence;
240               dataset()->padded_shapes_[component_index];
288               &batch_component, dataset()->padding_values_[component_index]));
328           input_impl_ = dataset()->input_->MakeIterator(prefix());