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,
|