Searched refs:dataset_ctx (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | dataset_test_base.cc | 797 std::unique_ptr<OpKernelContext> dataset_ctx; in MakeDataset() local 800 &dataset_ctx_params, &dataset_ctx, in MakeDataset() 804 std::move(dataset_ctx), std::move(created_tensors), dataset_base); in MakeDataset() 813 std::unique_ptr<OpKernelContext>* dataset_ctx) { in RunDatasetOp() argument 836 dataset_ctx_params, dataset_ctx)); in RunDatasetOp() 837 TF_RETURN_IF_ERROR(RunOpKernel(dataset_kernel->get(), dataset_ctx->get())); in RunDatasetOp() 845 std::unique_ptr<OpKernelContext>* dataset_ctx, in MakeDataset() argument 850 dataset_ctx)); in MakeDataset() 852 DCHECK_EQ((*dataset_ctx)->num_outputs(), 1); in MakeDataset() 853 TF_RETURN_IF_ERROR(GetDatasetFromContext(dataset_ctx->get(), 0, dataset)); in MakeDataset() [all …]
|
D | dataset_test_base.h | 726 std::unique_ptr<OpKernelContext>* dataset_ctx); 732 std::unique_ptr<OpKernelContext>* dataset_ctx,
|