Searched refs:dataset_kernel_ctx_ (Results 1 – 2 of 2) sorted by relevance
153 dataset_kernel_.get(), &inputs, &dataset_kernel_ctx_)); in CreateDatasetFromTestCase()155 dataset_kernel_ctx_.get(), &dataset_)); in CreateDatasetFromTestCase()162 CreateIteratorContext(dataset_kernel_ctx_.get(), &iterator_ctx_)); in CreateIteratorFromTestCase()169 std::unique_ptr<OpKernelContext> dataset_kernel_ctx_; member in tensorflow::data::__anon516b62740111::ConcatenateDatasetOpTestHelper
138 &dataset_kernel_ctx_)); in CreateDatasetFromTestCase()140 dataset_kernel_ctx_.get(), &dataset_)); in CreateDatasetFromTestCase()147 CreateIteratorContext(dataset_kernel_ctx_.get(), &iterator_ctx_)); in CreateIteratorFromTestCase()154 std::unique_ptr<OpKernelContext> dataset_kernel_ctx_; member in tensorflow::data::__anon6eabbe220111::ZipDatasetOpTestHelper