Searched refs:dataset_base (Results 1 – 1 of 1) sorted by relevance
809 DatasetBase* dataset_base; in MakeDataset() local816 &created_tensors, &dataset_base)); in MakeDataset()819 std::move(dataset_ctx), std::move(created_tensors), dataset_base); in MakeDataset()967 DatasetBase* dataset_base; in MakeDatasetTensor() local975 CreateDataset(dataset_kernel.get(), dataset_ctx.get(), &dataset_base)); in MakeDatasetTensor()978 StoreDatasetInVariantTensor(dataset_base, &dataset_tensor)); in MakeDatasetTensor()