Searched refs:UnaryDatasetOpKernel (Results 1 – 25 of 44) sorted by relevance
12
25 class PrefetchDatasetOp : public UnaryDatasetOpKernel {28 : UnaryDatasetOpKernel(ctx) {} in PrefetchDatasetOp()
32 class OptimizeDatasetOp : public UnaryDatasetOpKernel {35 : UnaryDatasetOpKernel(ctx), in OptimizeDatasetOp()
32 class FilterByLastComponentDatasetOp : public UnaryDatasetOpKernel {35 : UnaryDatasetOpKernel(ctx), in FilterByLastComponentDatasetOp()
24 class TakeDatasetOp : public UnaryDatasetOpKernel {27 : UnaryDatasetOpKernel(ctx) {} in TakeDatasetOp()
27 class ShardDatasetOp : public UnaryDatasetOpKernel {30 : UnaryDatasetOpKernel(ctx) {} in ShardDatasetOp()
31 class ModelDatasetOp : public UnaryDatasetOpKernel {34 : UnaryDatasetOpKernel(ctx) { in ModelDatasetOp()
26 class SkipDatasetOp : public UnaryDatasetOpKernel {29 : UnaryDatasetOpKernel(ctx) {} in SkipDatasetOp()
27 class BatchDatasetOp : public UnaryDatasetOpKernel {30 : UnaryDatasetOpKernel(ctx), in BatchDatasetOp()
26 class RepeatDatasetOp : public UnaryDatasetOpKernel {29 : UnaryDatasetOpKernel(ctx) {} in RepeatDatasetOp()
30 class FlatMapDatasetOp : public UnaryDatasetOpKernel {33 : UnaryDatasetOpKernel(ctx), in FlatMapDatasetOp()
34 class FilterDatasetOp : public UnaryDatasetOpKernel {41 : UnaryDatasetOpKernel(ctx) { in FilterDatasetOp()
37 class ParallelMapDatasetOp : public UnaryDatasetOpKernel {40 : UnaryDatasetOpKernel(ctx) { in ParallelMapDatasetOp()
26 class SleepDatasetOp : public UnaryDatasetOpKernel {28 using UnaryDatasetOpKernel::UnaryDatasetOpKernel;
123 class ThreadPoolDatasetOp : public UnaryDatasetOpKernel {126 : UnaryDatasetOpKernel(ctx) {} in ThreadPoolDatasetOp()233 class MaxIntraOpParallelismDatasetOp : public UnaryDatasetOpKernel {236 : UnaryDatasetOpKernel(ctx) {} in MaxIntraOpParallelismDatasetOp()333 class PrivateThreadPoolDatasetOp : public UnaryDatasetOpKernel {336 : UnaryDatasetOpKernel(ctx) {} in PrivateThreadPoolDatasetOp()
37 class LatencyStatsDatasetOp : public UnaryDatasetOpKernel {40 : UnaryDatasetOpKernel(ctx) {} in LatencyStatsDatasetOp()147 class BytesProducedStatsDatasetOp : public UnaryDatasetOpKernel {150 : UnaryDatasetOpKernel(ctx) {} in BytesProducedStatsDatasetOp()
24 class RebatchDatasetOp : public UnaryDatasetOpKernel {27 : UnaryDatasetOpKernel(ctx), in RebatchDatasetOp()
24 class AutoShardDatasetOp : public UnaryDatasetOpKernel {27 : UnaryDatasetOpKernel(ctx), in AutoShardDatasetOp()
25 class NonSerializableDatasetOp : public UnaryDatasetOpKernel {28 : UnaryDatasetOpKernel(ctx) { in NonSerializableDatasetOp()
26 class IgnoreErrorsDatasetOp : public UnaryDatasetOpKernel {29 : UnaryDatasetOpKernel(ctx) {} in IgnoreErrorsDatasetOp()
27 class AssertNextDatasetOp : public UnaryDatasetOpKernel {30 : UnaryDatasetOpKernel(ctx) { in AssertNextDatasetOp()
74 class SetStatsAggregatorDatasetOp : public UnaryDatasetOpKernel {77 : UnaryDatasetOpKernel(ctx) {} in SetStatsAggregatorDatasetOp()
27 class UniqueDatasetOp : public UnaryDatasetOpKernel {30 : UnaryDatasetOpKernel(ctx) {} in UniqueDatasetOp()
27 class UnbatchDatasetOp : public UnaryDatasetOpKernel {30 : UnaryDatasetOpKernel(ctx) {} in UnbatchDatasetOp()
33 class TakeWhileDatasetOp : public UnaryDatasetOpKernel {40 : UnaryDatasetOpKernel(ctx) { in TakeWhileDatasetOp()
23 class BigtableLookupDatasetOp : public UnaryDatasetOpKernel {25 using UnaryDatasetOpKernel::UnaryDatasetOpKernel;