Searched refs:dataset_ctx_ (Results 1 – 2 of 2) sorted by relevance
775 &dataset_ctx_, &tensors_, &dataset_)); in Initialize()776 TF_RETURN_IF_ERROR(CreateIteratorContext(dataset_ctx_.get(), &iterator_ctx_)); in Initialize()886 &tensors_, &dataset_ctx_)); in RunDatasetOp()887 for (int i = 0; i < dataset_ctx_->num_outputs(); ++i) { in RunDatasetOp()888 outputs->emplace_back(*dataset_ctx_->mutable_output(i)); in RunDatasetOp()
780 std::unique_ptr<OpKernelContext> dataset_ctx_; variable