Lines Matching refs:dataset
15 #include "tensorflow/core/framework/dataset.h"
37 // Transfers ownership of `dataset` to `*this`.
38 explicit DatasetVariantWrapper(DatasetBase* dataset) : dataset_(dataset) {}
76 const GraphDatasetBase* dataset,
81 const string& op_type_name = dataset->op_name();
90 opts->WithAttr("output_shapes", dataset->output_shapes())));
94 opts->WithAttr("output_types", dataset->output_dtypes())));
217 "Dataset tensor must be a scalar of dtype DT_VARIANT.");
222 return errors::InvalidArgument("Tensor must be a Dataset object.");
226 return errors::Internal("Read uninitialized Dataset variant.");
231 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) {
235 "Dataset tensor must be a scalar of dtype DT_VARIANT.");
237 tensor->scalar<Variant>()() = DatasetVariantWrapper(dataset);
242 DatasetBase* dataset = nullptr;
243 MakeDataset(ctx, &dataset);
247 OP_REQUIRES_OK(ctx, StoreDatasetInVariantTensor(dataset, output));
