Browse Source

add variable layernormfuison for lite

tags/v1.1.1
xuanyue 5 years ago
parent
commit
0a023d77d5
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/core/base/core_ops.h

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

@@ -493,6 +493,7 @@ inline const PrimitivePtr kPrimActivation = std::make_shared<Primitive>("Activat
inline const PrimitivePtr kPrimTopKFusion = std::make_shared<Primitive>("TopKFusion");
inline const PrimitivePtr kPrimTileFusion = std::make_shared<Primitive>("TileFusion");
inline const PrimitivePtr kPrimReduceFusion = std::make_shared<Primitive>("ReduceFusion");
inline const PrimitivePtr kPrimLayerNormFusion = std::make_shared<Primitive>("LayerNormFusion");

class DoSignaturePrimitive : public Primitive {
public:


Loading…
Cancel
Save