| 357 | } |
| 358 | |
| 359 | Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) { |
| 360 | if (!(tensor->dtype() == DT_VARIANT && |
| 361 | TensorShapeUtils::IsScalar(tensor->shape()))) { |
| 362 | return errors::InvalidArgument( |
| 363 | "Dataset tensor must be a scalar of dtype DT_VARIANT."); |
| 364 | } |
| 365 | tensor->scalar<Variant>()() = DatasetVariantWrapper(dataset); |
| 366 | return Status::OK(); |
| 367 | } |
| 368 | |
| 369 | Status DatasetBase::DatasetGraphDefBuilder::AddInputDataset( |
| 370 | SerializationContext* ctx, const DatasetBase* dataset, Node** output) { |