Searched refs:dataset_kernel_ (Results 1 – 2 of 2) sorted by relevance
151 &dataset_kernel_)); in CreateDatasetFromTestCase()153 dataset_kernel_.get(), &inputs, &dataset_kernel_ctx_)); in CreateDatasetFromTestCase()154 TF_RETURN_IF_ERROR(CreateDataset(dataset_kernel_.get(), in CreateDatasetFromTestCase()168 std::unique_ptr<OpKernel> dataset_kernel_; member in tensorflow::data::__anon516b62740111::ConcatenateDatasetOpTestHelper
136 inputs.size(), &dataset_kernel_)); in CreateDatasetFromTestCase()137 TF_RETURN_IF_ERROR(CreateZipDatasetContext(dataset_kernel_.get(), &inputs, in CreateDatasetFromTestCase()139 TF_RETURN_IF_ERROR(CreateDataset(dataset_kernel_.get(), in CreateDatasetFromTestCase()153 std::unique_ptr<OpKernel> dataset_kernel_; member in tensorflow::data::__anon6eabbe220111::ZipDatasetOpTestHelper