Home
last modified time | relevance | path

Searched refs:dataset_kernel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc795 std::unique_ptr<OpKernel> dataset_kernel; in MakeDataset() local
799 TF_RETURN_IF_ERROR(MakeDataset(dataset_params, &dataset_kernel, in MakeDataset()
803 std::move(dataset_kernel), std::move(dataset_ctx_params), in MakeDataset()
810 std::unique_ptr<OpKernel>* dataset_kernel, in RunDatasetOp() argument
834 TF_RETURN_IF_ERROR(MakeDatasetOpKernel(dataset_params, dataset_kernel)); in RunDatasetOp()
835 TF_RETURN_IF_ERROR(CreateDatasetContext(dataset_kernel->get(), &inputs, in RunDatasetOp()
837 TF_RETURN_IF_ERROR(RunOpKernel(dataset_kernel->get(), dataset_ctx->get())); in RunDatasetOp()
843 std::unique_ptr<OpKernel>* dataset_kernel, in MakeDataset() argument
848 TF_RETURN_IF_ERROR(RunDatasetOp(dataset_params, dataset_kernel, in MakeDataset()
895 std::unique_ptr<OpKernel>* dataset_kernel) { in MakeDatasetOpKernel() argument
[all …]
Ddataset_test_base.h723 std::unique_ptr<OpKernel>* dataset_kernel,
730 std::unique_ptr<OpKernel>* dataset_kernel,
738 std::unique_ptr<OpKernel>* dataset_kernel);