Lines Matching refs:Dataset
18 #include "tensorflow/core/kernels/data/dataset.h"
25 // This op defines a `Dataset` that passes through its input elements and
46 *output = new Dataset(ctx, input, std::move(tag));
50 class Dataset : public GraphDatasetBase {
52 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, string tag)
57 ~Dataset() override { input_->Unref(); }
72 string DebugString() override { return "LatencyStatsDatasetOp::Dataset"; }
86 class Iterator : public DatasetIterator<Dataset> {
89 : DatasetIterator<Dataset>(params),
90 input_impl_(params.dataset->input_->MakeIterator(params.prefix)) {}
102 dataset()->tag_, {static_cast<double>(end - start)});
140 *output = new Dataset(ctx, input, std::move(tag));
144 class Dataset : public GraphDatasetBase {
146 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, string tag)
151 ~Dataset() override { input_->Unref(); }
167 return "BytesProducedStatsDatasetOp::Dataset";
182 class Iterator : public DatasetIterator<Dataset> {
185 : DatasetIterator<Dataset>(params),
186 input_impl_(params.dataset->input_->MakeIterator(params.prefix)) {}
200 dataset()->tag_, {static_cast<double>(total_bytes)});
