Searched refs:dataset_base (Results 1 – 1 of 1) sorted by relevance
794 DatasetBase* dataset_base; in MakeDataset() local801 &created_tensors, &dataset_base)); in MakeDataset()804 std::move(dataset_ctx), std::move(created_tensors), dataset_base); in MakeDataset()935 DatasetBase* dataset_base; in MakeDatasetTensor() local943 CreateDataset(dataset_kernel.get(), dataset_ctx.get(), &dataset_base)); in MakeDatasetTensor()946 StoreDatasetInVariantTensor(dataset_base, &dataset_tensor)); in MakeDatasetTensor()