Searched refs:ShuffleAndRepeatDatasetOp (Results 1 – 3 of 3) sorted by relevance
58 class ShuffleAndRepeatDatasetOp : public ShuffleDatasetOpBase {63 explicit ShuffleAndRepeatDatasetOp(OpKernelConstruction* ctx);
53 ShuffleAndRepeatDatasetOp::kDatasetType;54 /* static */ constexpr const char* const ShuffleAndRepeatDatasetOp::kCount;747 class ShuffleAndRepeatDatasetOp::Dataset : public ShuffleDatasetBase {802 class ShuffleAndRepeatDatasetOp::DatasetV2 : public ShuffleDatasetBase {866 ShuffleAndRepeatDatasetOp::ShuffleAndRepeatDatasetOp(OpKernelConstruction* ctx) in ShuffleAndRepeatDatasetOp() function in tensorflow::data::ShuffleAndRepeatDatasetOp880 void ShuffleAndRepeatDatasetOp::MakeDataset(OpKernelContext* ctx, in MakeDataset()938 *output = new ShuffleAndRepeatDatasetOp::DatasetV2( in MakeDataset()982 ShuffleAndRepeatDatasetOp);985 ShuffleAndRepeatDatasetOp);
64 input_names->emplace_back(ShuffleAndRepeatDatasetOp::kCount); in GetInputNames()82 return ShuffleAndRepeatDatasetOp::kDatasetType; in dataset_type()392 name_utils::OpName(ShuffleAndRepeatDatasetOp::kDatasetType)}}; in DATASET_NODE_NAME_TEST_P()