|
|
|
@@ -15,6 +15,7 @@ |
|
|
|
*/ |
|
|
|
|
|
|
|
#include <memory> |
|
|
|
#include <utility> |
|
|
|
#include "src/ops/ops_utils.h" |
|
|
|
|
|
|
|
#ifdef PRIMITIVE_WRITEABLE |
|
|
|
@@ -22,7 +23,7 @@ |
|
|
|
|
|
|
|
namespace mindspore { |
|
|
|
namespace lite { |
|
|
|
schema::PrimitiveT *GetPrimitiveT(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> GetPrimitiveT(const AnfNodePtr &node) { |
|
|
|
auto prim = GetValueNode<std::shared_ptr<Primitive>>(node); |
|
|
|
if (prim == nullptr) { |
|
|
|
MS_LOG(DEBUG) << "primitive is nullptr"; |
|
|
|
@@ -44,723 +45,723 @@ schema::PrimitiveT *GetPrimitiveT(const AnfNodePtr &node) { |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
schema::PrimitiveT *AbsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AbsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Abs>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AbsGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AbsGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AbsGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ActivationPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ActivationPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Activation>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ActivationGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ActivationGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ActivationGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AdamPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AdamPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Adam>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AdderFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AdderFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AdderFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AddFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AddFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AddFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AddGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AddGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AddGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AddNPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AddNPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AddN>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AllPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AllPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::All>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ApplyMomentumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ApplyMomentumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ApplyMomentum>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ArgMaxFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ArgMaxFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ArgMaxFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ArgMinFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ArgMinFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ArgMinFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AssertPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AssertPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Assert>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AssignPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AssignPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Assign>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AssignAddPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AssignAddPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AssignAdd>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AudioSpectrogramPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AudioSpectrogramPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AudioSpectrogram>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AvgPoolFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AvgPoolFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AvgPoolFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *AvgPoolGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> AvgPoolGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AvgPoolGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *BatchNormPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> BatchNormPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BatchNorm>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *BatchToSpacePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> BatchToSpacePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BatchToSpace>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *BatchToSpaceNDPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> BatchToSpaceNDPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BatchToSpaceND>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *BiasAddPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> BiasAddPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BiasAdd>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *BiasAddGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> BiasAddGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BiasAddGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *BNGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> BNGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BatchNormGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *BroadcastToPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> BroadcastToPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BroadcastTo>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CastPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CastPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Cast>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CeilPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CeilPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Ceil>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ClipPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ClipPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Clip>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ConcatPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ConcatPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Concat>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ConstantOfShapePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ConstantOfShapePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ConstantOfShape>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ControlDependPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ControlDependPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ControlDepend>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *Conv2DBackpropFilterFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> Conv2DBackpropFilterFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Conv2DBackpropFilterFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *Conv2DBackpropInputFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> Conv2DBackpropInputFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Conv2DBackpropInputFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *Conv2DFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> Conv2DFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Conv2DFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *Conv2dTransposeFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> Conv2dTransposeFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Conv2dTransposeFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CosPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CosPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Cos>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CropPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CropPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Crop>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CropAndResizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CropAndResizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::CropAndResize>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CustomExtractFeaturesPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CustomExtractFeaturesPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::CustomExtractFeatures>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CustomNormalizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CustomNormalizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::CustomNormalize>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *CustomPredictPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CustomPredictPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::CustomPredict>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *DependPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> DependPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Depend>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *DepthToSpacePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> DepthToSpacePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::DepthToSpace>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *DetectionPostProcessPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> DetectionPostProcessPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::DetectionPostProcess>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *DivFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> DivFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::DivFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *DivGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> DivGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::DivGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *DropoutPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> DropoutPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Dropout>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *DropoutGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> DropoutGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::DropoutGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *GRUPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> GRUPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::GRU>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *EltwisePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> EltwisePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Eltwise>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *EluPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> EluPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Elu>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *EmbeddingLookupFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> EmbeddingLookupFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::EmbeddingLookupFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *EqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> EqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Equal>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ExpandDimsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ExpandDimsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ExpandDims>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ExpFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ExpFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ExpFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FftImagPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FftImagPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::FftImag>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FftRealPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FftRealPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::FftReal>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FillPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FillPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Fill>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FlattenPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FlattenPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Flatten>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FlattenGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FlattenGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::FlattenGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FloorPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FloorPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Floor>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FloorDivPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FloorDivPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::FloorDiv>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FloorModPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FloorModPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::FloorMod>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FullConnectionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FullConnectionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::FullConnection>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *FusedBatchNormPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> FusedBatchNormPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::FusedBatchNorm>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *GatherPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> GatherPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Gather>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *GatherNdPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> GatherNdPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::GatherNd>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *GreaterPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> GreaterPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Greater>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *GreaterEqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> GreaterEqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::GreaterEqual>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *HashtableLookupPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> HashtableLookupPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::HashtableLookup>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *InstanceNormPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> InstanceNormPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::InstanceNorm>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *InvertPermutationPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> InvertPermutationPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::InvertPermutation>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LayerNormFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LayerNormFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LayerNormFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LayerNormGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LayerNormGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LayerNormGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LeakyReluPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LeakyReluPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LeakyRelu>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LessPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LessPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Less>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LessEqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LessEqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LessEqual>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LogPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LogPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Log>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LogGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LogGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LogGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LogicalAndPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LogicalAndPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LogicalAnd>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LogicalNotPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LogicalNotPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LogicalNot>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LogicalOrPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LogicalOrPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LogicalOr>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LrnPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LrnPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LRN>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LpNormalizationPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LpNormalizationPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LpNormalization>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LshProjectionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LshProjectionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LshProjection>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *LSTMPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LSTMPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LSTM>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *L2NormalizeFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> L2NormalizeFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::L2NormalizeFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MatMulPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MatMulPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::MatMul>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MaximumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MaximumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Maximum>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MaximumGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MaximumGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::MaximumGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MaxPoolFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MaxPoolFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::MaxPoolFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MaxPoolGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MaxPoolGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::MaxPoolGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MergePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MergePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Merge>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MfccPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MfccPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Mfcc>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MinimumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MinimumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Minimum>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MinimumGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MinimumGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::MinimumGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ModPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ModPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Mod>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MulFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MulFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::MulFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *MulGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> MulGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::MulGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *NegPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> NegPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Neg>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *NegGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> NegGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::NegGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *NotEqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> NotEqualPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::NotEqual>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *NonMaxSuppressionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> NonMaxSuppressionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::NonMaxSuppression>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *OneHotPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> OneHotPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::OneHot>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *OnesLikePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> OnesLikePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::OnesLike>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *PadFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> PadFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::PadFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *PartialFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> PartialFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::PartialFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *PowerGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> PowerGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::PowerGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *PowFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> PowFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::PowFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *PReLUFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> PReLUFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::PReLUFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *QuantDTypeCastPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> QuantDTypeCastPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::QuantDTypeCast>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RangePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RangePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Range>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RandomStandardNormalPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RandomStandardNormalPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::RandomStandardNormal>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RankPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RankPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Rank>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RealDivPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RealDivPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::RealDiv>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ReciprocalPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ReciprocalPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Reciprocal>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ReduceFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ReduceFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ReduceFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ReshapePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ReshapePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Reshape>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ResizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ResizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Resize>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ResizeGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ResizeGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ResizeGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ReverseV2PrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ReverseV2PrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ReverseV2>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ReverseSequencePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ReverseSequencePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ReverseSequence>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RfftPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RfftPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Rfft>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ROIPoolingPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ROIPoolingPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ROIPooling>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RoundPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RoundPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Round>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RsqrtPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RsqrtPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Rsqrt>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *RsqrtGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> RsqrtGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::RsqrtGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ScaleFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ScaleFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ScaleFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ScatterNdPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ScatterNdPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ScatterNd>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SelectPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SelectPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Select>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SGDPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SGDPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SGD>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ShapePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ShapePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Shape>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SigmoidCrossEntropyWithLogitsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SigmoidCrossEntropyWithLogitsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SigmoidCrossEntropyWithLogits>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SigmoidCrossEntropyWithLogitsGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SigmoidCrossEntropyWithLogitsGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SigmoidCrossEntropyWithLogitsGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SinPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SinPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Sin>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SizePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Size>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SkipGramPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SkipGramPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SkipGram>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SliceFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SliceFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SliceFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SmoothL1LossPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SmoothL1LossPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SmoothL1Loss>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SmoothL1LossGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SmoothL1LossGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SmoothL1LossGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SoftmaxPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SoftmaxPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Softmax>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SoftmaxCrossEntropyWithLogitsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SoftmaxCrossEntropyWithLogitsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SoftmaxCrossEntropyWithLogits>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SpaceToBatchPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SpaceToBatchPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SpaceToBatch>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SpaceToBatchNDPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SpaceToBatchNDPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SpaceToBatchND>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SpaceToDepthPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SpaceToDepthPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SpaceToDepth>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SparseSoftmaxCrossEntropyWithLogitsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SparseSoftmaxCrossEntropyWithLogitsPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SparseSoftmaxCrossEntropyWithLogits>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SparseToDensePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SparseToDensePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SparseToDense>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SplitPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SplitPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Split>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SqrtPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SqrtPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Sqrt>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SqrtGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SqrtGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SqrtGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SquarePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SquarePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Square>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SquaredDifferencePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SquaredDifferencePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SquaredDifference>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SqueezePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SqueezePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Squeeze>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *StackPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> StackPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Stack>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *StridedSlicePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> StridedSlicePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::StridedSlice>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *StridedSliceGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> StridedSliceGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::StridedSliceGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SubFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SubFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SubFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SubGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SubGradPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SubGrad>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *SwitchPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SwitchPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Switch>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TensorListFromTensorPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TensorListFromTensorPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::TensorListFromTensor>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TensorListGetItemPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TensorListGetItemPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::TensorListGetItem>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TensorListReservePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TensorListReservePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::TensorListReserve>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TensorListSetItemPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TensorListSetItemPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::TensorListSetItem>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TensorListStackPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TensorListStackPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::TensorListStack>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TileFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TileFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::TileFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TopKFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TopKFusionPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::TopKFusion>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *TransposePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> TransposePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Transpose>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *UniquePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> UniquePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Unique>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *UnstackPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> UnstackPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Unstack>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *UnsortedSegmentSumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> UnsortedSegmentSumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::UnsortedSegmentSum>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *UnsqueezePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> UnsqueezePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Unsqueeze>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *WherePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> WherePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Where>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ZerosLikePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ZerosLikePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ZerosLike>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ErfPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> ErfPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Erf>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
|
|
|
|
schema::PrimitiveT *SplicePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> SplicePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Splice>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
|
|
|
|
schema::PrimitiveT *LogSoftmaxPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> LogSoftmaxPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::LogSoftmax>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
|
|
|
|
schema::PrimitiveT *CallPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CallPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Call>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
|
|
|
|
schema::PrimitiveT *CumSumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CumSumPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::CumSum>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
@@ -982,9 +983,9 @@ RegistryMSOps g_LogSoftmaxPrimitiveCreatorRegistry("LogSoftmax", LogSoftmaxPrimi |
|
|
|
RegistryMSOps g_CallPrimitiveCreatorRegistry("call", CallPrimitiveCreator); |
|
|
|
RegistryMSOps g_CumSumPrimitiveCreatorRegistry("CumSum", CumSumPrimitiveCreator); |
|
|
|
|
|
|
|
schema::PrimitiveT *CustomPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
std::unique_ptr<schema::PrimitiveT> CustomPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Custom>>(node); |
|
|
|
auto *schema_op = new (std::nothrow) schema::CustomT(); |
|
|
|
auto schema_op = std::make_unique<schema::CustomT>(); |
|
|
|
if (schema_op == nullptr) { |
|
|
|
return nullptr; |
|
|
|
} |
|
|
|
@@ -994,21 +995,21 @@ schema::PrimitiveT *CustomPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
if (ms_primc->GetAttr("attr") != nullptr) { |
|
|
|
auto attr_map = ms_primc->get_attr(); |
|
|
|
for (const auto &attr_item : attr_map) { |
|
|
|
auto *attr = new (std::nothrow) schema::AttributeT(); |
|
|
|
auto attr = std::make_unique<schema::AttributeT>(); |
|
|
|
if (attr == nullptr) { |
|
|
|
return nullptr; |
|
|
|
} |
|
|
|
attr->name = attr_item.first; |
|
|
|
attr->data = attr_item.second; |
|
|
|
schema_op->attr.emplace_back(attr); |
|
|
|
schema_op->attr.emplace_back(std::move(attr)); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
auto *prim = new (std::nothrow) schema::PrimitiveT(); |
|
|
|
auto prim = std::make_unique<schema::PrimitiveT>(); |
|
|
|
if (prim == nullptr) { |
|
|
|
return nullptr; |
|
|
|
} |
|
|
|
prim->value.value = schema_op; |
|
|
|
prim->value.value = schema_op.release(); |
|
|
|
prim->value.type = schema::PrimitiveType_Custom; |
|
|
|
return prim; |
|
|
|
} |
|
|
|
|