|
|
|
@ -28,6 +28,31 @@
|
|
|
|
|
namespace mindspore {
|
|
|
|
|
namespace dataset {
|
|
|
|
|
|
|
|
|
|
constexpr char kBarrierOp[] = "BarrierOp";
|
|
|
|
|
constexpr char kBatchOp[] = "BatchOp";
|
|
|
|
|
constexpr char kBucketBatchByLengthOp[] = "BucketBatchByLengthOp";
|
|
|
|
|
constexpr char kBuildSentencePieceVocabOp[] = "BuildSentencePieceVocabOp";
|
|
|
|
|
constexpr char kBuildVocabOp[] = "BuildVocabOp";
|
|
|
|
|
constexpr char kCacheBase[] = "CacheBase";
|
|
|
|
|
constexpr char kCacheLookupOp[] = "CacheLookupOp";
|
|
|
|
|
constexpr char kCacheMergeOp[] = "CacheMergeOp";
|
|
|
|
|
constexpr char kCacheOp[] = "CacheOp";
|
|
|
|
|
constexpr char kConcatOp[] = "ConcatOp";
|
|
|
|
|
constexpr char kDatasetOp[] = "DatasetOp";
|
|
|
|
|
constexpr char kDeviceQueueOp[] = "DeviceQueueOp";
|
|
|
|
|
constexpr char kEpochCtrlOp[] = "EpochCtrlOp";
|
|
|
|
|
constexpr char kFilterOp[] = "FilterOp";
|
|
|
|
|
constexpr char kMapOp[] = "MapOp";
|
|
|
|
|
constexpr char kParallelOp[] = "ParallelOp";
|
|
|
|
|
constexpr char kPipelineOp[] = "PipelineOp";
|
|
|
|
|
constexpr char kProjectOp[] = "ProjectOp";
|
|
|
|
|
constexpr char kRenameOp[] = "RenameOp";
|
|
|
|
|
constexpr char kRepeatOp[] = "RepeatOp";
|
|
|
|
|
constexpr char kShuffleOp[] = "ShuffleOp";
|
|
|
|
|
constexpr char kSkipOp[] = "SkipOp";
|
|
|
|
|
constexpr char kTakeOp[] = "TakeOp";
|
|
|
|
|
constexpr char kZipOp[] = "ZipOp";
|
|
|
|
|
|
|
|
|
|
// Forward declare
|
|
|
|
|
class ExecutionTree;
|
|
|
|
|
|
|
|
|
@ -292,7 +317,7 @@ class DatasetOp : public std::enable_shared_from_this<DatasetOp> {
|
|
|
|
|
|
|
|
|
|
/// Op name getter
|
|
|
|
|
/// \return Name of the current Op
|
|
|
|
|
virtual std::string Name() const { return "DatasetOp"; }
|
|
|
|
|
virtual std::string Name() const = 0;
|
|
|
|
|
|
|
|
|
|
/// Execution Tree getter
|
|
|
|
|
/// \return Pointer to the ExecutionTree the current op belongs to, no ownership
|
|
|
|
|