Browse Source

ONNX converter: second batch of changes

feature/build-system-rewrite
Alexander Malyshev 4 years ago
parent
commit
1937dbd785
2 changed files with 626 additions and 221 deletions
  1. +624
    -221
      mindspore/ccsrc/transform/express_ir/onnx_exporter.cc
  2. +2
    -0
      mindspore/core/base/core_ops.h

+ 624
- 221
mindspore/ccsrc/transform/express_ir/onnx_exporter.cc
File diff suppressed because it is too large
View File


+ 2
- 0
mindspore/core/base/core_ops.h View File

@@ -530,6 +530,8 @@ MS_CORE_API inline const PrimitivePtr kPrimApplyProximalGradientDescent =
MS_CORE_API inline const PrimitivePtr kPrimSparseApplyRMSProp = std::make_shared<Primitive>("SparseApplyRMSProp");
MS_CORE_API inline const PrimitivePtr kPrimApplyKerasMomentum = std::make_shared<Primitive>("ApplyKerasMomentum");
MS_CORE_API inline const PrimitivePtr kPrimLARSUpdate = std::make_shared<Primitive>("LARSUpdate");
MS_CORE_API inline const PrimitivePtr kPrimBoundingBoxDecode = std::make_shared<Primitive>("BoundingBoxDecode");
MS_CORE_API inline const PrimitivePtr kPrimROIAlign = std::make_shared<Primitive>("ROIAlign");
MS_CORE_API inline const PrimitivePtr kPrimApplyAddSign = std::make_shared<Primitive>("ApplyAddSign");
MS_CORE_API inline const PrimitivePtr kPrimApplyAdagrad = std::make_shared<Primitive>("ApplyAdagrad");
MS_CORE_API inline const PrimitivePtr kPrimApplyAdadelta = std::make_shared<Primitive>("ApplyAdadelta");


Loading…
Cancel
Save