Searched refs:PrivateThreadPoolDatasetOp (Results 1 – 3 of 3) sorted by relevance
39 PrivateThreadPoolDatasetOp::kDatasetType;40 /* static */ constexpr const char* const PrivateThreadPoolDatasetOp::kDatasetOp;409 class PrivateThreadPoolDatasetOp::Dataset : public DatasetBase {528 void PrivateThreadPoolDatasetOp::MakeDatasetFromOptions(OpKernelContext* ctx, in MakeDatasetFromOptions()536 {PrivateThreadPoolDatasetOp::kDatasetType, in MakeDatasetFromOptions()537 PrivateThreadPoolDatasetOp::kDatasetOp})), in MakeDatasetFromOptions()541 void PrivateThreadPoolDatasetOp::MakeDataset(OpKernelContext* ctx, in MakeDataset()561 PrivateThreadPoolDatasetOp);564 PrivateThreadPoolDatasetOp);
49 class PrivateThreadPoolDatasetOp : public UnaryDatasetOpKernel {59 explicit PrivateThreadPoolDatasetOp(OpKernelConstruction* ctx) in PrivateThreadPoolDatasetOp() function
59 experimental::PrivateThreadPoolDatasetOp::MakeDatasetFromOptions( in MakeDatasetHelper()