Searched refs:PrefetchDatasetOp (Results 1 – 3 of 3) sorted by relevance
25 class PrefetchDatasetOp : public UnaryDatasetOpKernel {27 explicit PrefetchDatasetOp(OpKernelConstruction* ctx) in PrefetchDatasetOp() function
36 class PrefetchDatasetOp::Dataset : public DatasetBase {382 void PrefetchDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()400 PrefetchDatasetOp);407 PrefetchDatasetOp);
241 data::PrefetchDatasetOp); \