From 1bce8d63376fa9b1e68556d1d69292f97dfe5532 Mon Sep 17 00:00:00 2001 From: xutianchun Date: Thu, 11 Mar 2021 21:01:57 +0800 Subject: [PATCH] fix LSTM --- mindspore/core/base/core_ops.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/core/base/core_ops.h b/mindspore/core/base/core_ops.h index 56dfbf235e..7c4edfdc09 100644 --- a/mindspore/core/base/core_ops.h +++ b/mindspore/core/base/core_ops.h @@ -193,7 +193,7 @@ inline const PrimitivePtr kPrimSoftmax = std::make_shared("Softmax"); inline const PrimitivePtr kPrimSparseSoftmaxCrossEntropy = std::make_shared("SparseSoftmaxCrossEntropy"); inline const PrimitivePtr kPrimLogSoftmax = std::make_shared("LogSoftmax"); inline const PrimitivePtr kPrimLogSoftmaxGrad = std::make_shared("LogSoftmaxGrad"); -inline const PrimitivePtr kPrimLstm = std::make_shared("Lstm"); +inline const PrimitivePtr kPrimLstm = std::make_shared("LSTM"); inline const PrimitivePtr kPrimTan = std::make_shared("Tan"); inline const PrimitivePtr kPrimAtan2 = std::make_shared("Atan2"); inline const PrimitivePtr kPrimAtan = std::make_shared("Atan");