Home
last modified time | relevance | path

Searched refs:ShuffleDatasetOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.h40 class ShuffleDatasetOp : public ShuffleDatasetOpBase {
44 explicit ShuffleDatasetOp(OpKernelConstruction* ctx);
Dshuffle_dataset_op.cc54 /* static */ constexpr const char* const ShuffleDatasetOp::kDatasetType;
519 class ShuffleDatasetOp::Dataset : public ShuffleDatasetBase {
577 class ShuffleDatasetOp::DatasetV2 : public ShuffleDatasetBase {
631 class ShuffleDatasetOp::DatasetV3 : public ShuffleDatasetBase {
693 ShuffleDatasetOp::ShuffleDatasetOp(OpKernelConstruction* ctx) in ShuffleDatasetOp() function in tensorflow::data::ShuffleDatasetOp
709 void ShuffleDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
757 *output = new ShuffleDatasetOp::DatasetV3(ctx, input, buffer_size, count, in MakeDataset()
787 new ShuffleDatasetOp::DatasetV2(ctx, input, buffer_size, count, manager, in MakeDataset()
819 *output = new ShuffleDatasetOp::Dataset(ctx, input, buffer_size, count, in MakeDataset()
1051 ShuffleDatasetOp);
[all …]
Dshuffle_dataset_op_test.cc88 return ShuffleDatasetOp::kDatasetType; in dataset_type()
394 ShuffleDatasetOp::kDatasetType)}, in DATASET_NODE_NAME_TEST_P()
453 ShuffleDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()