From 0a023d77d58490124b2fbb0eff9ca4d62f4e8427 Mon Sep 17 00:00:00 2001 From: xuanyue Date: Mon, 25 Jan 2021 10:16:09 +0800 Subject: [PATCH] add variable layernormfuison for lite --- mindspore/core/base/core_ops.h | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/core/base/core_ops.h b/mindspore/core/base/core_ops.h index 71533f5680..fc0c06527d 100644 --- a/mindspore/core/base/core_ops.h +++ b/mindspore/core/base/core_ops.h @@ -493,6 +493,7 @@ inline const PrimitivePtr kPrimActivation = std::make_shared("Activat inline const PrimitivePtr kPrimTopKFusion = std::make_shared("TopKFusion"); inline const PrimitivePtr kPrimTileFusion = std::make_shared("TileFusion"); inline const PrimitivePtr kPrimReduceFusion = std::make_shared("ReduceFusion"); +inline const PrimitivePtr kPrimLayerNormFusion = std::make_shared("LayerNormFusion"); class DoSignaturePrimitive : public Primitive { public: