data_set.h

#pragma once
#include "paddle/fluid/framework/data_set.h"

namespace paddle {
namespace framework {

class FeedMultiSlotDataset : public MultiSlotDataset {
 public:
  FeedMultiSlotDataset() {}
  virtual void MergeByInsId();
  virtual void CreatePreLoadReaders();
  virtual ~FeedMultiSlotDataset() {}
};

}  // end namespace framework
}  // end namespace paddle