Lines Matching refs:dataset
21 #include "tensorflow/core/kernels/data/dataset.h"
89 *output = new Dataset(
96 class Dataset : public GraphDatasetBase {
98 Dataset(OpKernelContext* ctx, const DatasetBase* input,
119 ~Dataset() override { input_->Unref(); }
134 string DebugString() override { return "GroupByWindowDatasetOp::Dataset"; }
198 class Iterator : public DatasetIterator<Dataset> {
201 : DatasetIterator<Dataset>(params),
202 input_impl_(params.dataset->input_->MakeIterator(params.prefix)) {}
226 // Iterate through the input dataset until we get a full
238 dataset()->captured_key_func_->RunWithBorrowedArgs(
254 TF_RETURN_IF_ERROR(dataset()->captured_window_size_func_->Run(
451 groups_[key], dataset()->input_->output_dtypes(),
452 dataset()->input_->output_shapes(), &group_dataset));
464 TF_RETURN_IF_ERROR(dataset()->captured_reduce_func_->Run(
475 // Retrieve the dataset that was created in `f`.
481 // Create an iterator for the dataset that was returned by `f`.
