Searched refs:DatasetVariantWrapper (Results 1 – 1 of 1) sorted by relevance
39 class DatasetVariantWrapper { class41 DatasetVariantWrapper() : dataset_(nullptr) {} in DatasetVariantWrapper() function in tensorflow::data::__anon86842c2d0111::DatasetVariantWrapper44 explicit DatasetVariantWrapper(DatasetBase* dataset) : dataset_(dataset) {} in DatasetVariantWrapper() function in tensorflow::data::__anon86842c2d0111::DatasetVariantWrapper46 DatasetVariantWrapper(const DatasetVariantWrapper& other) in DatasetVariantWrapper() function in tensorflow::data::__anon86842c2d0111::DatasetVariantWrapper51 ~DatasetVariantWrapper() { in ~DatasetVariantWrapper()341 const DatasetVariantWrapper* wrapper = variant.get<DatasetVariantWrapper>(); in GetDatasetFromVariantTensor()358 tensor->scalar<Variant>()() = DatasetVariantWrapper(dataset); in StoreDatasetInVariantTensor()