| @@ -56,7 +56,6 @@ AbstractBasePtr AbsInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Abs, prim::kPrimAbs, AbsInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAbs, Abs); | REGISTER_PRIMITIVE_C(kNameAbs, Abs); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -81,8 +81,6 @@ AbstractBasePtr AdamInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| const std::vector<AbstractBasePtr> &input_args) { | const std::vector<AbstractBasePtr> &input_args) { | ||||
| return std::make_shared<abstract::AbstractTensor>(AdamInfer(primitive, input_args)); | return std::make_shared<abstract::AbstractTensor>(AdamInfer(primitive, input_args)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Adam, prim::kPrimAdam, AdamInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAdam, Adam); | REGISTER_PRIMITIVE_C(kNameAdam, Adam); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -50,7 +50,6 @@ AbstractBasePtr AddInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Add, prim::kPrimAdd, AddInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAdd, Add); | REGISTER_PRIMITIVE_C(kNameAdd, Add); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -63,8 +63,6 @@ AbstractBasePtr AddNInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| return std::make_shared<abstract::AbstractTensor>(TypeIdToType(infer_type), | return std::make_shared<abstract::AbstractTensor>(TypeIdToType(infer_type), | ||||
| std::make_shared<abstract::Shape>(element0_shape)); | std::make_shared<abstract::Shape>(element0_shape)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(AddN, prim::kPrimAddN, AddNInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAddN, AddN); | REGISTER_PRIMITIVE_C(kNameAddN, AddN); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -80,7 +80,6 @@ AbstractBasePtr ApplyMomentumInfer(const abstract::AnalysisEnginePtr &, const Pr | |||||
| auto element = g_type_tensor->element(); | auto element = g_type_tensor->element(); | ||||
| return std::make_shared<abstract::AbstractTensor>(element, v_shape); | return std::make_shared<abstract::AbstractTensor>(element, v_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(ApplyMomentum, prim::kPrimApplyMomentum, ApplyMomentumInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameApplyMomentum, ApplyMomentum); | REGISTER_PRIMITIVE_C(kNameApplyMomentum, ApplyMomentum); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -67,7 +67,6 @@ AbstractBasePtr ArgMaxInfer(const abstract::AnalysisEnginePtr &, const Primitive | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(ArgMax, prim::kPrimArgMax, ArgMaxInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameArgMax, ArgMax); | REGISTER_PRIMITIVE_C(kNameArgMax, ArgMax); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -67,7 +67,6 @@ AbstractBasePtr ArgMinInfer(const abstract::AnalysisEnginePtr &, const Primitive | |||||
| return std::make_shared<abstract::AbstractTensor>(x_dtype, std::make_shared<abstract::Shape>(out_shape)); | return std::make_shared<abstract::AbstractTensor>(x_dtype, std::make_shared<abstract::Shape>(out_shape)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(ArgMin, prim::kPrimArgMin, ArgMinInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameArgMin, ArgMin); | REGISTER_PRIMITIVE_C(kNameArgMin, ArgMin); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -46,7 +46,6 @@ AbstractBasePtr AsinInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape()); | return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Asin, prim::kPrimAsin, AsinInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAsin, Asin); | REGISTER_PRIMITIVE_C(kNameAsin, Asin); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -71,8 +71,6 @@ AbstractBasePtr AssertInfer(const abstract::AnalysisEnginePtr &, const Primitive | |||||
| } | } | ||||
| return std::make_shared<abstract::AbstractTensor>(TypeIdToType(kNumberTypeInt32), output_shape); | return std::make_shared<abstract::AbstractTensor>(TypeIdToType(kNumberTypeInt32), output_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Assert, prim::kPrimAssert, AssertInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAssert, Assert); | REGISTER_PRIMITIVE_C(kNameAssert, Assert); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -47,7 +47,6 @@ AbstractBasePtr AssignAddInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(AssignAdd, prim::kPrimAssignAdd, AssignAddInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAssignAdd, AssignAdd); | REGISTER_PRIMITIVE_C(kNameAssignAdd, AssignAdd); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -44,7 +44,6 @@ AbstractBasePtr AtanInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape()); | return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Atan, prim::kPrimAtan, AtanInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAtan, Atan); | REGISTER_PRIMITIVE_C(kNameAtan, Atan); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -118,8 +118,6 @@ AbstractBasePtr AudioSpectrogramInfer(const abstract::AnalysisEnginePtr &, const | |||||
| return std::make_shared<abstract::AbstractTensor>(AudioSpectrogramInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(AudioSpectrogramInferType(primitive, input_args), | ||||
| AudioSpectrogramInferShape(primitive, input_args)->shape()); | AudioSpectrogramInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(AudioSpectrogram, prim::kPrimAudioSpectrogram, AudioSpectrogramInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAudioSpectrogram, AudioSpectrogram); | REGISTER_PRIMITIVE_C(kNameAudioSpectrogram, AudioSpectrogram); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -146,7 +146,6 @@ AbstractBasePtr AvgPoolInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(AvgPool, prim::kPrimAvgPool, AvgPoolInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAvgPool, AvgPool); | REGISTER_PRIMITIVE_C(kNameAvgPool, AvgPool); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -134,7 +134,6 @@ AbstractBasePtr BatchNormInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| AbstractBasePtrList output = {output0, output1, output2, output3, output3}; | AbstractBasePtrList output = {output0, output1, output2, output3, output3}; | ||||
| return std::make_shared<abstract::AbstractTuple>(output); | return std::make_shared<abstract::AbstractTuple>(output); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BatchNorm, prim::kPrimBatchNorm, BatchNormInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBatchNorm, BatchNorm); | REGISTER_PRIMITIVE_C(kNameBatchNorm, BatchNorm); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -109,8 +109,6 @@ AbstractBasePtr BatchNormFoldInfer(const abstract::AnalysisEnginePtr &, const Pr | |||||
| AbstractBasePtrList output1 = {output, output, output, output}; | AbstractBasePtrList output1 = {output, output, output, output}; | ||||
| return std::make_shared<abstract::AbstractTuple>(output1); | return std::make_shared<abstract::AbstractTuple>(output1); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BatchNormFold, prim::kPrimBatchNormFold, BatchNormFoldInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBatchNormFold, BatchNormFold); | REGISTER_PRIMITIVE_C(kNameBatchNormFold, BatchNormFold); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -75,7 +75,6 @@ AbstractBasePtr BatchToSpaceInfer(const abstract::AnalysisEnginePtr &, const Pri | |||||
| ret->set_shape(std::make_shared<abstract::Shape>(out_shape)); | ret->set_shape(std::make_shared<abstract::Shape>(out_shape)); | ||||
| return ret; | return ret; | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BatchToSpace, prim::kPrimBatchToSpace, BatchToSpaceInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBatchToSpace, BatchToSpace); | REGISTER_PRIMITIVE_C(kNameBatchToSpace, BatchToSpace); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -103,7 +103,6 @@ AbstractBasePtr BatchToSpaceNDInfer(const abstract::AnalysisEnginePtr &, const P | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BatchToSpaceND, prim::kPrimBatchToSpaceND, BatchToSpaceNDInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBatchToSpaceND, BatchToSpaceND); | REGISTER_PRIMITIVE_C(kNameBatchToSpaceND, BatchToSpaceND); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -74,7 +74,6 @@ AbstractBasePtr BiasAddInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| BiasAddInferShape(primitive, input_args)); | BiasAddInferShape(primitive, input_args)); | ||||
| } | } | ||||
| // Add | // Add | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BiasAdd, prim::kPrimBiasAdd, BiasAddInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBiasAdd, BiasAdd); | REGISTER_PRIMITIVE_C(kNameBiasAdd, BiasAdd); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -87,8 +87,6 @@ AbstractBasePtr BinaryCrossEntropyInfer(const abstract::AnalysisEnginePtr &, con | |||||
| return std::make_shared<abstract::AbstractTensor>(BinaryCrossEntroyInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(BinaryCrossEntroyInferType(primitive, input_args), | ||||
| BinaryCrossEntroyInferShape(primitive, input_args)->shape()); | BinaryCrossEntroyInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BinaryCrossEntropy, prim::kPrimBinaryCrossEntropy, BinaryCrossEntropyInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBinaryCrossEntropy, BinaryCrossEntropy); | REGISTER_PRIMITIVE_C(kNameBinaryCrossEntropy, BinaryCrossEntropy); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -64,7 +64,6 @@ AbstractBasePtr BroadcastInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| } | } | ||||
| return std::make_shared<abstract::AbstractTensor>(x_type, in_shape); | return std::make_shared<abstract::AbstractTensor>(x_type, in_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Broadcast, prim::kPrimBroadcast, BroadcastInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBroadcast, Broadcast); | REGISTER_PRIMITIVE_C(kNameBroadcast, Broadcast); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -81,8 +81,6 @@ AbstractBasePtr BroadcastToInfer(const abstract::AnalysisEnginePtr &, const Prim | |||||
| return std::make_shared<abstract::AbstractTensor>(BroadcastToInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(BroadcastToInferType(primitive, input_args), | ||||
| BroadcastToInferShape(primitive, input_args)->shape()); | BroadcastToInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BroadcastTo, prim::kPrimBroadcastTo, BroadcastToInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBroadcastTo, BroadcastTo); | REGISTER_PRIMITIVE_C(kNameBroadcastTo, BroadcastTo); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -43,7 +43,6 @@ AbstractBasePtr CeilInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| MS_EXCEPTION_IF_NULL(data_type); | MS_EXCEPTION_IF_NULL(data_type); | ||||
| return std::make_shared<abstract::AbstractTensor>(data_type, x_shape); | return std::make_shared<abstract::AbstractTensor>(data_type, x_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Ceil, prim::kPrimCeil, CeilInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameCeil, Ceil); | REGISTER_PRIMITIVE_C(kNameCeil, Ceil); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -78,8 +78,6 @@ AbstractBasePtr ConcatInfer(const abstract::AnalysisEnginePtr &, const Primitive | |||||
| return std::make_shared<abstract::AbstractTensor>(TypeIdToType(infer_type), | return std::make_shared<abstract::AbstractTensor>(TypeIdToType(infer_type), | ||||
| std::make_shared<abstract::Shape>(ret_shape)); | std::make_shared<abstract::Shape>(ret_shape)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Concat, prim::kPrimConcat, ConcatInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameConcat, Concat); | REGISTER_PRIMITIVE_C(kNameConcat, Concat); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -52,7 +52,6 @@ AbstractBasePtr ConstantInfer(const abstract::AnalysisEnginePtr &, const Primiti | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Constant, prim::kPrimConstant, ConstantInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameConstant, Constant); | REGISTER_PRIMITIVE_C(kNameConstant, Constant); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -61,7 +61,6 @@ AbstractBasePtr ConstantOfShapeInfer(const abstract::AnalysisEnginePtr &, const | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(ConstantOfShape, prim::kPrimConstantOfShape, ConstantOfShapeInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameConstantOfShape, ConstantOfShape); | REGISTER_PRIMITIVE_C(kNameConstantOfShape, ConstantOfShape); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -232,8 +232,6 @@ AbstractBasePtr Conv2dInfer(const abstract::AnalysisEnginePtr &, const Primitive | |||||
| return std::make_shared<abstract::AbstractTensor>(Conv2dInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(Conv2dInferType(primitive, input_args), | ||||
| Conv2dInferShape(primitive, input_args)->shape()); | Conv2dInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Conv2D, prim::kPrimConv2D, Conv2dInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameConv2D, Conv2D); | REGISTER_PRIMITIVE_C(kNameConv2D, Conv2D); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -192,8 +192,6 @@ AbstractBasePtr Conv2dTransposeInfer(const abstract::AnalysisEnginePtr &, const | |||||
| return std::make_shared<abstract::AbstractTensor>(Conv2dTransposeInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(Conv2dTransposeInferType(primitive, input_args), | ||||
| Conv2dTransposeInferShape(primitive, input_args)->shape()); | Conv2dTransposeInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Conv2dTranspose, prim::kPrimConv2DTranspose, Conv2dTransposeInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameConv2dTranspose, Conv2dTranspose); | REGISTER_PRIMITIVE_C(kNameConv2dTranspose, Conv2dTranspose); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -50,7 +50,6 @@ AbstractBasePtr CosInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Cos, prim::kPrimCos, CosInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameCos, Cos); | REGISTER_PRIMITIVE_C(kNameCos, Cos); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -56,7 +56,6 @@ AbstractBasePtr CropInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| auto x_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); | auto x_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); | ||||
| return std::make_shared<abstract::AbstractTensor>(x_type, out_shape); | return std::make_shared<abstract::AbstractTensor>(x_type, out_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Crop, prim::kPrimCrop, CropInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameCrop, Crop); | REGISTER_PRIMITIVE_C(kNameCrop, Crop); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -50,7 +50,6 @@ AbstractBasePtr CustomExtractFeaturesInfer(const abstract::AnalysisEnginePtr &, | |||||
| AbstractBasePtrList output = {output0, output1}; | AbstractBasePtrList output = {output0, output1}; | ||||
| return std::make_shared<abstract::AbstractTuple>(output); | return std::make_shared<abstract::AbstractTuple>(output); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(CustomExtractFeatures, prim::kPrimCustomExtractFeatures, CustomExtractFeaturesInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameCustomExtractFeatures, CustomExtractFeatures); | REGISTER_PRIMITIVE_C(kNameCustomExtractFeatures, CustomExtractFeatures); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -62,8 +62,6 @@ AbstractBasePtr CustomNormalizeInfer(const abstract::AnalysisEnginePtr &, const | |||||
| return std::make_shared<abstract::AbstractTensor>(CustomNormalizeInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(CustomNormalizeInferType(primitive, input_args), | ||||
| CustomNormalizeInferShape(primitive, input_args)->shape()); | CustomNormalizeInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(CustomNormalize, prim::kPrimCustomNormalize, CustomNormalizeInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameCustomNormalize, CustomNormalize); | REGISTER_PRIMITIVE_C(kNameCustomNormalize, CustomNormalize); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -58,8 +58,6 @@ AbstractBasePtr CustomPredictInfer(const abstract::AnalysisEnginePtr &, const Pr | |||||
| AbstractBasePtrList output = {output0, output1}; | AbstractBasePtrList output = {output0, output1}; | ||||
| return std::make_shared<abstract::AbstractTuple>(output); | return std::make_shared<abstract::AbstractTuple>(output); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(CustomPredict, prim::kPrimCustomPredict, CustomPredictInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameCustomPredict, CustomPredict); | REGISTER_PRIMITIVE_C(kNameCustomPredict, CustomPredict); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -81,7 +81,6 @@ AbstractBasePtr DepthToSpaceInfer(const abstract::AnalysisEnginePtr &, const Pri | |||||
| ret->set_shape(std::make_shared<abstract::Shape>(out_shape)); | ret->set_shape(std::make_shared<abstract::Shape>(out_shape)); | ||||
| return ret; | return ret; | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(DepthToSpace, prim::kPrimDepthToSpace, DepthToSpaceInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameDepthToSpace, DepthToSpace); | REGISTER_PRIMITIVE_C(kNameDepthToSpace, DepthToSpace); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -168,7 +168,6 @@ AbstractBasePtr DetectionPostProcessInfer(const abstract::AnalysisEnginePtr &, c | |||||
| } | } | ||||
| return std::make_shared<abstract::AbstractTuple>(output); | return std::make_shared<abstract::AbstractTuple>(output); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(DetectionPostProcess, prim::kPrimDetectionPostProcess, DetectionPostProcessInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameDetectionPostProcess, DetectionPostProcess); | REGISTER_PRIMITIVE_C(kNameDetectionPostProcess, DetectionPostProcess); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -51,7 +51,6 @@ AbstractBasePtr DivInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Div, prim::kPrimDiv, DivInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameDiv, Div); | REGISTER_PRIMITIVE_C(kNameDiv, Div); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -64,7 +64,6 @@ AbstractBasePtr DropoutInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape()); | return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Dropout, prim::kPrimDropout, DropoutInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameDropout, Dropout); | REGISTER_PRIMITIVE_C(kNameDropout, Dropout); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -68,8 +68,6 @@ AbstractBasePtr EluInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Elu, prim::kPrimElu, EluInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameElu, Elu); | REGISTER_PRIMITIVE_C(kNameElu, Elu); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -80,8 +80,6 @@ AbstractBasePtr EmbeddingLookupInfer(const abstract::AnalysisEnginePtr &, const | |||||
| return std::make_shared<abstract::AbstractTensor>(params->element(), | return std::make_shared<abstract::AbstractTensor>(params->element(), | ||||
| std::make_shared<abstract::Shape>(shape, min_shape, max_shape)); | std::make_shared<abstract::Shape>(shape, min_shape, max_shape)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(EmbeddingLookup, prim::kPrimEmbeddingLookup, EmbeddingLookupInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameEmbeddingLookup, EmbeddingLookup); | REGISTER_PRIMITIVE_C(kNameEmbeddingLookup, EmbeddingLookup); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -52,7 +52,6 @@ AbstractBasePtr EqualInfer(const abstract::AnalysisEnginePtr &, const PrimitiveP | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Equal, prim::kPrimEqual, EqualInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameEqual, Equal); | REGISTER_PRIMITIVE_C(kNameEqual, Equal); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -48,7 +48,6 @@ AbstractBasePtr ExpInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Exp, prim::kPrimExp, ExpInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameExp, Exp); | REGISTER_PRIMITIVE_C(kNameExp, Exp); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -55,7 +55,6 @@ AbstractBasePtr ExpandDimsInfer(const abstract::AnalysisEnginePtr &, const Primi | |||||
| CheckAndConvertUtils::CheckSubClass("x_type", x_type, valid_x_type, prim_name); | CheckAndConvertUtils::CheckSubClass("x_type", x_type, valid_x_type, prim_name); | ||||
| return std::make_shared<abstract::AbstractTensor>(x_type, out_shape); | return std::make_shared<abstract::AbstractTensor>(x_type, out_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(ExpandDims, prim::kPrimExpandDims, ExpandDimsInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameExpandDims, ExpandDims); | REGISTER_PRIMITIVE_C(kNameExpandDims, ExpandDims); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -85,7 +85,6 @@ AbstractBasePtr FakeQuantWithMinMaxVarsInfer(const abstract::AnalysisEnginePtr & | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(FakeQuantWithMinMaxVars, prim::kPrimFakeQuantWithMinMaxVars, FakeQuantWithMinMaxVarsInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFakeQuantWithMinMaxVars, FakeQuantWithMinMaxVars); | REGISTER_PRIMITIVE_C(kNameFakeQuantWithMinMaxVars, FakeQuantWithMinMaxVars); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -69,9 +69,6 @@ AbstractBasePtr FakeQuantWithMinMaxVarsPerChannelInfer(const abstract::AnalysisE | |||||
| MS_EXCEPTION_IF_NULL(data_type); | MS_EXCEPTION_IF_NULL(data_type); | ||||
| return std::make_shared<abstract::AbstractTensor>(data_type, x_shape); | return std::make_shared<abstract::AbstractTensor>(data_type, x_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(FakeQuantWithMinMaxVarsPerChannel, prim::kPrimFakeQuantWithMinMaxVarsPerChannel, | |||||
| FakeQuantWithMinMaxVarsPerChannelInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFakeQuantWithMinMaxVarsPerChannel, FakeQuantWithMinMaxVarsPerChannel); | REGISTER_PRIMITIVE_C(kNameFakeQuantWithMinMaxVarsPerChannel, FakeQuantWithMinMaxVarsPerChannel); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -45,7 +45,6 @@ AbstractBasePtr FftImagInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(FftImag, prim::kPrimFftImag, FftImagInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFftImag, FftImag); | REGISTER_PRIMITIVE_C(kNameFftImag, FftImag); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -37,7 +37,6 @@ AbstractBasePtr FftRealInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| out_shape.pop_back(); | out_shape.pop_back(); | ||||
| return std::make_shared<abstract::AbstractTensor>(out_dtype, std::make_shared<abstract::Shape>(out_shape)); | return std::make_shared<abstract::AbstractTensor>(out_dtype, std::make_shared<abstract::Shape>(out_shape)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(FftReal, prim::kPrimFftReal, FftRealInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFftReal, FftReal); | REGISTER_PRIMITIVE_C(kNameFftReal, FftReal); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -58,7 +58,6 @@ AbstractBasePtr FillInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| abs->set_value(tensor); | abs->set_value(tensor); | ||||
| return abs; | return abs; | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Fill, prim::kPrimFill, FillInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFill, Fill); | REGISTER_PRIMITIVE_C(kNameFill, Fill); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -53,7 +53,6 @@ AbstractBasePtr FlattenInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Flatten, prim::kPrimFlatten, FlattenInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFlatten, Flatten); | REGISTER_PRIMITIVE_C(kNameFlatten, Flatten); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -54,7 +54,6 @@ AbstractBasePtr FloorInfer(const abstract::AnalysisEnginePtr &, const PrimitiveP | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Floor, prim::kPrimFloor, FloorInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFloor, Floor); | REGISTER_PRIMITIVE_C(kNameFloor, Floor); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -62,7 +62,6 @@ AbstractBasePtr AddFusionInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(AddFusion, prim::kPrimAddFusion, AddFusionInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAddFusion, AddFusion); | REGISTER_PRIMITIVE_C(kNameAddFusion, AddFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -44,7 +44,6 @@ ActivationType AdderFusion::get_activation_type() const { | |||||
| auto value_ptr = GetAttr(kActivationType); | auto value_ptr = GetAttr(kActivationType); | ||||
| return ActivationType(GetValue<int64_t>(value_ptr)); | return ActivationType(GetValue<int64_t>(value_ptr)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameAdderFusion, AdderFusion); | REGISTER_PRIMITIVE_C(kNameAdderFusion, AdderFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -105,7 +105,6 @@ AbstractBasePtr AvgPoolFusionInfer(const abstract::AnalysisEnginePtr &, const Pr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(AvgPoolFusion, prim::kPrimAvgPool, AvgPoolFusionInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAvgPoolFusion, AvgPoolFusion); | REGISTER_PRIMITIVE_C(kNameAvgPoolFusion, AvgPoolFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -60,7 +60,6 @@ int64_t Conv2DBackpropFilterFusion::get_in_channel() const { | |||||
| auto value_ptr = GetAttr(kInChannel); | auto value_ptr = GetAttr(kInChannel); | ||||
| return GetValue<int64_t>(value_ptr); | return GetValue<int64_t>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameConv2DBackpropFilterFusion, Conv2DBackpropFilterFusion); | REGISTER_PRIMITIVE_C(kNameConv2DBackpropFilterFusion, Conv2DBackpropFilterFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -110,7 +110,6 @@ AbstractBasePtr FullConnectionInfer(const abstract::AnalysisEnginePtr &, const P | |||||
| auto input0_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); | auto input0_type = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); | ||||
| return std::make_shared<abstract::AbstractTensor>(input0_type, out_shape); | return std::make_shared<abstract::AbstractTensor>(input0_type, out_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(FullConnection, prim::kPrimFullConnection, FullConnectionInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFullConnection, FullConnection); | REGISTER_PRIMITIVE_C(kNameFullConnection, FullConnection); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -60,7 +60,6 @@ AbstractBasePtr PowFusionInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(PowFusion, prim::kPrimPowFusion, PowFusionInfer); | |||||
| REGISTER_PRIMITIVE_C(kNamePowFusion, PowFusion); | REGISTER_PRIMITIVE_C(kNamePowFusion, PowFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -42,7 +42,6 @@ std::vector<float> PReLUFusion::get_slope() const { | |||||
| auto value_ptr = GetAttr(kSlope); | auto value_ptr = GetAttr(kSlope); | ||||
| return GetValue<std::vector<float>>(value_ptr); | return GetValue<std::vector<float>>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNamePReLUFusion, PReLUFusion); | REGISTER_PRIMITIVE_C(kNamePReLUFusion, PReLUFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -67,8 +67,6 @@ void ReduceFusion::Init(const bool keep_dims, const ReduceMode mode, const bool | |||||
| this->set_reduce_to_end(reduce_to_end); | this->set_reduce_to_end(reduce_to_end); | ||||
| this->set_coeff(coeff); | this->set_coeff(coeff); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(ReduceFusion, prim::kPrimReduceMean, ReduceInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameReduceFusion, ReduceFusion); | REGISTER_PRIMITIVE_C(kNameReduceFusion, ReduceFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -63,8 +63,6 @@ AbstractBasePtr SliceFusionInfer(const abstract::AnalysisEnginePtr &, const Prim | |||||
| } | } | ||||
| return std::make_shared<abstract::AbstractTensor>(data_type, size); | return std::make_shared<abstract::AbstractTensor>(data_type, size); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(SliceFusion, prim::kPrimSliceFusion, SliceFusionInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameSliceFusion, SliceFusion); | REGISTER_PRIMITIVE_C(kNameSliceFusion, SliceFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -28,7 +28,6 @@ std::vector<int64_t> TileFusion::get_dims() const { | |||||
| auto value_ptr = GetAttr(kDims); | auto value_ptr = GetAttr(kDims); | ||||
| return GetValue<std::vector<int64_t>>(value_ptr); | return GetValue<std::vector<int64_t>>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameTileFusion, TileFusion); | REGISTER_PRIMITIVE_C(kNameTileFusion, TileFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -39,7 +39,6 @@ int64_t TopKFusion::get_largest() const { | |||||
| auto value_ptr = GetAttr(kLargest); | auto value_ptr = GetAttr(kLargest); | ||||
| return GetValue<int64_t>(value_ptr); | return GetValue<int64_t>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameTopKFusion, TopKFusion); | REGISTER_PRIMITIVE_C(kNameTopKFusion, TopKFusion); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -42,7 +42,6 @@ AbstractBasePtr GatherInfer(const abstract::AnalysisEnginePtr &, const Primitive | |||||
| return std::make_shared<abstract::AbstractTensor>(x_type, index_shape); | return std::make_shared<abstract::AbstractTensor>(x_type, index_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Gather, prim::kPrimGather, GatherInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameGather, Gather); | REGISTER_PRIMITIVE_C(kNameGather, Gather); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -67,7 +67,6 @@ AbstractBasePtr GatherNdInfer(const abstract::AnalysisEnginePtr &, const Primiti | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(GatherNd, prim::kPrimGatherND, GatherNdInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameGatherNd, GatherNd); | REGISTER_PRIMITIVE_C(kNameGatherNd, GatherNd); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -51,8 +51,6 @@ AbstractBasePtr GeLUInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| return std::make_shared<abstract::AbstractTensor>(GeLUInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(GeLUInferType(primitive, input_args), | ||||
| GeLUInferShape(primitive, input_args)->shape()); | GeLUInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(GeLU, prim::kPrimGeLU, GeLUInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameGeLU, GeLU); | REGISTER_PRIMITIVE_C(kNameGeLU, GeLU); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -32,7 +32,6 @@ AbstractBasePtr AvgPoolGradInfer(const abstract::AnalysisEnginePtr &, const Prim | |||||
| return std::make_shared<abstract::AbstractTensor>(element, origin_input_shape); | return std::make_shared<abstract::AbstractTensor>(element, origin_input_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(AvgPoolGrad, prim::kPrimAvgPoolGrad, AvgPoolGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameAvgPoolGrad, AvgPoolGrad); | REGISTER_PRIMITIVE_C(kNameAvgPoolGrad, AvgPoolGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -66,8 +66,6 @@ AbstractBasePtr BatchNormGradInfer(const abstract::AnalysisEnginePtr &, const Pr | |||||
| AbstractBasePtrList rets = {dx, dscale, dscale, reserve_1, reserve_2}; | AbstractBasePtrList rets = {dx, dscale, dscale, reserve_1, reserve_2}; | ||||
| return std::make_shared<abstract::AbstractTuple>(rets); | return std::make_shared<abstract::AbstractTuple>(rets); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BatchNormGrad, prim::kPrimBatchNormGrad, BatchNormGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBatchNormGrad, BatchNormGrad); | REGISTER_PRIMITIVE_C(kNameBatchNormGrad, BatchNormGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -46,7 +46,6 @@ AbstractBasePtr BiasGradInfer(const abstract::AnalysisEnginePtr &, const Primiti | |||||
| return std::make_shared<abstract::AbstractTensor>(intype, inshape); | return std::make_shared<abstract::AbstractTensor>(intype, inshape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BiasGrad, prim::kPrimBiasGrad, BiasGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBiasGrad, BiasGrad); | REGISTER_PRIMITIVE_C(kNameBiasGrad, BiasGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -73,8 +73,6 @@ AbstractBasePtr BinaryCrossEntropyGradInfer(const abstract::AnalysisEnginePtr &, | |||||
| return std::make_shared<abstract::AbstractTensor>(BinaryCrossEntroyGradInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(BinaryCrossEntroyGradInferType(primitive, input_args), | ||||
| BinaryCrossEntroyGradInferShape(primitive, input_args)->shape()); | BinaryCrossEntroyGradInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(BinaryCrossEntropyGrad, prim::kPrimBinaryCrossEntropyGrad, BinaryCrossEntropyGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameBinaryCrossEntropyGrad, BinaryCrossEntropyGrad); | REGISTER_PRIMITIVE_C(kNameBinaryCrossEntropyGrad, BinaryCrossEntropyGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -39,7 +39,6 @@ float BNGrad::get_momentum() const { | |||||
| auto value_ptr = this->GetAttr(kMomentum); | auto value_ptr = this->GetAttr(kMomentum); | ||||
| return GetValue<float>(value_ptr); | return GetValue<float>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameBNGrad, BNGrad); | REGISTER_PRIMITIVE_C(kNameBNGrad, BNGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -146,8 +146,6 @@ AbstractBasePtr Conv2DBackpropFilterInfer(const abstract::AnalysisEnginePtr &, c | |||||
| return std::make_shared<abstract::AbstractTensor>(Conv2DBackpropFilterInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(Conv2DBackpropFilterInferType(primitive, input_args), | ||||
| Conv2DBackpropFilterInferShape(primitive, input_args)->shape()); | Conv2DBackpropFilterInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Conv2DBackpropFilter, prim::kPrimConv2DBackpropFilter, Conv2DBackpropFilterInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameConv2DBackpropFilter, Conv2DBackpropFilter); | REGISTER_PRIMITIVE_C(kNameConv2DBackpropFilter, Conv2DBackpropFilter); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -196,7 +196,6 @@ std::vector<int64_t> Conv2DBackpropInput::get_pad_list() const { | |||||
| auto value_ptr = GetAttr(kPadList); | auto value_ptr = GetAttr(kPadList); | ||||
| return GetValue<std::vector<int64_t>>(value_ptr); | return GetValue<std::vector<int64_t>>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Conv2DBackpropInput, prim::kPrimConv2DBackpropInput, Conv2DBackpropInputInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameConv2DBackpropInput, Conv2DBackpropInput); | REGISTER_PRIMITIVE_C(kNameConv2DBackpropInput, Conv2DBackpropInput); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -132,7 +132,6 @@ bool DeConv2DGradFilter::get_has_bias() const { | |||||
| auto value_ptr = GetAttr(kHasBias); | auto value_ptr = GetAttr(kHasBias); | ||||
| return GetValue<bool>(value_ptr); | return GetValue<bool>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameDeConv2DGradFilter, DeConv2DGradFilter); | REGISTER_PRIMITIVE_C(kNameDeConv2DGradFilter, DeConv2DGradFilter); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -63,8 +63,6 @@ AbstractBasePtr DropoutGradInfer(const abstract::AnalysisEnginePtr &, const Prim | |||||
| return std::make_shared<abstract::AbstractTensor>(DropoutGradInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(DropoutGradInferType(primitive, input_args), | ||||
| DropoutGradInferShape(primitive, input_args)->shape()); | DropoutGradInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(DropoutGrad, prim::kPrimDropoutGrad, DropoutGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameDropoutGrad, DropoutGrad); | REGISTER_PRIMITIVE_C(kNameDropoutGrad, DropoutGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -35,7 +35,6 @@ AbstractBasePtr FlattenGradInfer(const abstract::AnalysisEnginePtr &, const Prim | |||||
| ret->set_shape(std::make_shared<abstract::Shape>(out_shape)); | ret->set_shape(std::make_shared<abstract::Shape>(out_shape)); | ||||
| return ret; | return ret; | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(FlattenGrad, prim::kPrimFlattenGrad, FlattenGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameFlattenGrad, FlattenGrad); | REGISTER_PRIMITIVE_C(kNameFlattenGrad, FlattenGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -161,7 +161,6 @@ AbstractBasePtr GroupConv2DGradInputInfer(const abstract::AnalysisEnginePtr &, c | |||||
| return std::make_shared<abstract::AbstractTensor>(type, shape); | return std::make_shared<abstract::AbstractTensor>(type, shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(GroupConv2DGradInput, prim::kPrimGroupConv2DGradInput, GroupConv2DGradInputInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameGroupConv2DGradInput, GroupConv2DGradInput); | REGISTER_PRIMITIVE_C(kNameGroupConv2DGradInput, GroupConv2DGradInput); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -60,8 +60,6 @@ AbstractBasePtr MaxPoolGradInfer(const abstract::AnalysisEnginePtr &, const Prim | |||||
| auto element = tensor_type->element(); | auto element = tensor_type->element(); | ||||
| return std::make_shared<abstract::AbstractTensor>(element, x1_shape); | return std::make_shared<abstract::AbstractTensor>(element, x1_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(MaxPoolGrad, prim::kPrimMaxPoolGrad, MaxPoolGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameMaxPoolGrad, MaxPoolGrad); | REGISTER_PRIMITIVE_C(kNameMaxPoolGrad, MaxPoolGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -37,7 +37,6 @@ bool MaximumGrad::get_grad_y() const { | |||||
| auto value_ptr = GetAttr(kGradY); | auto value_ptr = GetAttr(kGradY); | ||||
| return GetValue<bool>(value_ptr); | return GetValue<bool>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameMaximumGrad, MaximumGrad); | REGISTER_PRIMITIVE_C(kNameMaximumGrad, MaximumGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -37,7 +37,6 @@ bool MinimumGrad::get_grad_y() const { | |||||
| auto value_ptr = GetAttr(kGradY); | auto value_ptr = GetAttr(kGradY); | ||||
| return GetValue<bool>(value_ptr); | return GetValue<bool>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameMinimumGrad, MinimumGrad); | REGISTER_PRIMITIVE_C(kNameMinimumGrad, MinimumGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -19,7 +19,6 @@ | |||||
| namespace mindspore { | namespace mindspore { | ||||
| namespace ops { | namespace ops { | ||||
| std::vector<int64_t> PoolGrad::_grad_check_vector(std::string arg_name, std::vector<int64_t> arg_val, | std::vector<int64_t> PoolGrad::_grad_check_vector(std::string arg_name, std::vector<int64_t> arg_val, | ||||
| std::string op_name) { | std::string op_name) { | ||||
| std::vector<int64_t> ret; | std::vector<int64_t> ret; | ||||
| @@ -83,7 +82,6 @@ PadMode PoolGrad::get_pad_mode() const { | |||||
| auto value_ptr = GetAttr(kPadMode); | auto value_ptr = GetAttr(kPadMode); | ||||
| return PadMode(GetValue<int64_t>(value_ptr)); | return PadMode(GetValue<int64_t>(value_ptr)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNamePoolGrad, PoolGrad); | REGISTER_PRIMITIVE_C(kNamePoolGrad, PoolGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -58,9 +58,6 @@ AbstractBasePtr SigmoidCrossEntropyWithLogitsGradInfer(const abstract::AnalysisE | |||||
| return std::make_shared<abstract::AbstractTensor>(dout_type, x_shape); | return std::make_shared<abstract::AbstractTensor>(dout_type, x_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(SigmoidCrossEntropyWithLogitsGrad, prim::kPrimSigmoidCrossEntropyWithLogitsGrad, | |||||
| SigmoidCrossEntropyWithLogitsGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameSigmoidCrossEntropyWithLogitsGrad, SigmoidCrossEntropyWithLogitsGrad); | REGISTER_PRIMITIVE_C(kNameSigmoidCrossEntropyWithLogitsGrad, SigmoidCrossEntropyWithLogitsGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -63,7 +63,6 @@ AbstractBasePtr SmoothL1LossGradInfer(const abstract::AnalysisEnginePtr &, const | |||||
| return std::make_shared<abstract::AbstractTensor>(dloss_type, prediction); | return std::make_shared<abstract::AbstractTensor>(dloss_type, prediction); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(SmoothL1LossGrad, prim::kPrimSmoothL1LossGrad, SmoothL1LossGradInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameSmoothL1LossGrad, SmoothL1LossGrad); | REGISTER_PRIMITIVE_C(kNameSmoothL1LossGrad, SmoothL1LossGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -78,7 +78,6 @@ int64_t StridedSliceGrad::get_shrink_axis_mask() const { | |||||
| auto value_ptr = GetAttr(kShrinkAxisMask); | auto value_ptr = GetAttr(kShrinkAxisMask); | ||||
| return GetValue<int64_t>(value_ptr); | return GetValue<int64_t>(value_ptr); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameStridedSliceGrad, StridedSliceGrad); | REGISTER_PRIMITIVE_C(kNameStridedSliceGrad, StridedSliceGrad); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -111,7 +111,6 @@ GateOrderMode GRU::get_gate_order() const { | |||||
| auto value_ptr = this->GetAttr(kGateOrder); | auto value_ptr = this->GetAttr(kGateOrder); | ||||
| return GateOrderMode(GetValue<int64_t>(value_ptr)); | return GateOrderMode(GetValue<int64_t>(value_ptr)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_C(kNameGRU, GRU); | REGISTER_PRIMITIVE_C(kNameGRU, GRU); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -49,8 +49,6 @@ AbstractBasePtr HashtableLookupInfer(const abstract::AnalysisEnginePtr &, const | |||||
| } | } | ||||
| return std::make_shared<abstract::AbstractTuple>(output1); | return std::make_shared<abstract::AbstractTuple>(output1); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(HashtableLookup, prim::kPrimHashtableLookup, HashtableLookupInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameHashtableLookup, HashtableLookup); | REGISTER_PRIMITIVE_C(kNameHashtableLookup, HashtableLookup); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -59,7 +59,6 @@ AbstractBasePtr L2NormalizeInfer(const abstract::AnalysisEnginePtr &, const Prim | |||||
| } | } | ||||
| return input_args[0]->Broaden(); | return input_args[0]->Broaden(); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(L2Normalize, prim::kPrimL2Normalize, L2NormalizeInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameL2Normalize, L2Normalize); | REGISTER_PRIMITIVE_C(kNameL2Normalize, L2Normalize); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -51,7 +51,6 @@ AbstractBasePtr LeakyReluInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LeakyRelu, prim::kPrimLeakyRelu, LeakyReluInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLeakyRelu, LeakyRelu); | REGISTER_PRIMITIVE_C(kNameLeakyRelu, LeakyRelu); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -50,7 +50,6 @@ AbstractBasePtr LessInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Less, prim::kPrimLess, LessInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLess, Less); | REGISTER_PRIMITIVE_C(kNameLess, Less); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -51,7 +51,6 @@ AbstractBasePtr LessEqualInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LessEqual, prim::kPrimLessEqual, LessEqualInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLessEqual, LessEqual); | REGISTER_PRIMITIVE_C(kNameLessEqual, LessEqual); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -91,8 +91,6 @@ AbstractBasePtr LocalResponseNormalizationInfer(const abstract::AnalysisEnginePt | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LocalResponseNormalization, prim::kPrimLocalResponseNormalization, | |||||
| LocalResponseNormalizationInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLocalResponseNormalization, LocalResponseNormalization); | REGISTER_PRIMITIVE_C(kNameLocalResponseNormalization, LocalResponseNormalization); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -41,7 +41,6 @@ AbstractBasePtr LogInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Log, prim::kPrimLog, LogInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLog, Log); | REGISTER_PRIMITIVE_C(kNameLog, Log); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -55,7 +55,6 @@ AbstractBasePtr LogicalAndInfer(const abstract::AnalysisEnginePtr &, const Primi | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LogicalAnd, prim::kPrimLogicalAnd, LogicalAndInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLogicalAnd, LogicalAnd); | REGISTER_PRIMITIVE_C(kNameLogicalAnd, LogicalAnd); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -50,8 +50,6 @@ AbstractBasePtr LogicalNotInfer(const abstract::AnalysisEnginePtr &, const Primi | |||||
| return std::make_shared<abstract::AbstractTensor>(LogicalNotInferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(LogicalNotInferType(primitive, input_args), | ||||
| LogicalNotInferShape(primitive, input_args)->shape()); | LogicalNotInferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LogicalNot, prim::kPrimLogicalNot, LogicalNotInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLogicalNot, LogicalNot); | REGISTER_PRIMITIVE_C(kNameLogicalNot, LogicalNot); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -56,7 +56,6 @@ AbstractBasePtr LogicalOrInfer(const abstract::AnalysisEnginePtr &, const Primit | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LogicalOr, prim::kPrimLogicalOr, LogicalOrInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLogicalOr, LogicalOr); | REGISTER_PRIMITIVE_C(kNameLogicalOr, LogicalOr); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -102,7 +102,6 @@ AbstractBasePtr LrnInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Lrn, prim::kPrimLrn, LrnInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLrn, Lrn); | REGISTER_PRIMITIVE_C(kNameLrn, Lrn); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -64,8 +64,6 @@ AbstractBasePtr LshProjectionInfer(const abstract::AnalysisEnginePtr &, const Pr | |||||
| TypePtr infer_type = TypeIdToType(kNumberTypeInt32); | TypePtr infer_type = TypeIdToType(kNumberTypeInt32); | ||||
| return std::make_shared<abstract::AbstractTensor>(infer_type, out_shape); | return std::make_shared<abstract::AbstractTensor>(infer_type, out_shape); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LshProjection, prim::kPrimLshProjection, LshProjectionInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLshProjection, LshProjection); | REGISTER_PRIMITIVE_C(kNameLshProjection, LshProjection); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -169,8 +169,6 @@ AbstractBasePtr LstmInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| const std::vector<AbstractBasePtr> &input_args) { | const std::vector<AbstractBasePtr> &input_args) { | ||||
| return std::make_shared<abstract::AbstractTensor>(LstmInfer(primitive, input_args)); | return std::make_shared<abstract::AbstractTensor>(LstmInfer(primitive, input_args)); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(LSTM, prim::kPrimLstm, LstmInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameLSTM, LSTM); | REGISTER_PRIMITIVE_C(kNameLSTM, LSTM); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -95,7 +95,6 @@ AbstractBasePtr MatMulInfer(const abstract::AnalysisEnginePtr &, const Primitive | |||||
| } | } | ||||
| // Add | // Add | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(MatMul, prim::kPrimMatMul, MatMulInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameMatMul, MatMul); | REGISTER_PRIMITIVE_C(kNameMatMul, MatMul); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -80,7 +80,6 @@ AbstractBasePtr MatrixDiagInfer(const abstract::AnalysisEnginePtr &, const Primi | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(MatrixDiag, prim::kPrimMatrixDiag, MatrixDiagInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameMatrixDiag, MatrixDiag); | REGISTER_PRIMITIVE_C(kNameMatrixDiag, MatrixDiag); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -152,7 +152,6 @@ AbstractBasePtr MaxPoolInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(MaxPool, prim::kPrimMaxPool, MaxPoolInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameMaxPool, MaxPool); | REGISTER_PRIMITIVE_C(kNameMaxPool, MaxPool); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -48,7 +48,6 @@ AbstractBasePtr MaximumInfer(const abstract::AnalysisEnginePtr &, const Primitiv | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Maximum, prim::kPrimMaximum, MaximumInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameMaximum, Maximum); | REGISTER_PRIMITIVE_C(kNameMaximum, Maximum); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -52,8 +52,6 @@ AbstractBasePtr MergeInfer(const abstract::AnalysisEnginePtr &, const PrimitiveP | |||||
| AbstractBasePtrList output = {output1, output2}; | AbstractBasePtrList output = {output1, output2}; | ||||
| return std::make_shared<abstract::AbstractTuple>(output); | return std::make_shared<abstract::AbstractTuple>(output); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Merge, prim::kPrimMerge, MergeInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameMerge, Merge); | REGISTER_PRIMITIVE_C(kNameMerge, Merge); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -93,7 +93,6 @@ AbstractBasePtr MfccInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt | |||||
| return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), | ||||
| InferShape(primitive, input_args)->shape()); | InferShape(primitive, input_args)->shape()); | ||||
| } | } | ||||
| REGISTER_PRIMITIVE_EVAL_IMPL(Mfcc, prim::kPrimMfcc, MfccInfer); | |||||
| REGISTER_PRIMITIVE_C(kNameMfcc, Mfcc); | REGISTER_PRIMITIVE_C(kNameMfcc, Mfcc); | ||||
| } // namespace ops | } // namespace ops | ||||
| } // namespace mindspore | } // namespace mindspore | ||||