Searched refs:GeneratorDatasetOp (Results 1 – 3 of 3) sorted by relevance
31 class GeneratorDatasetOp::Dataset : public DatasetBase {155 GeneratorDatasetOp::GeneratorDatasetOp(OpKernelConstruction* ctx) in GeneratorDatasetOp() function in tensorflow::data::GeneratorDatasetOp164 void GeneratorDatasetOp::MakeDataset(OpKernelContext* ctx, in MakeDataset()186 GeneratorDatasetOp);191 GeneratorDatasetOp);
24 class GeneratorDatasetOp : public DatasetOpKernel {26 explicit GeneratorDatasetOp(OpKernelConstruction* ctx);
235 data::GeneratorDatasetOp); \