From db8bf4e589c5b89cee496f5ec53b2cec2db00e15 Mon Sep 17 00:00:00 2001
From: "dongduo5@huawei.com" <dongduo5@huawei.com>
Date: Mon, 12 Apr 2021 19:24:27 +0800
Subject: [PATCH] add fractal zn index

---
 .../format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc | 4 ++--
 ge/common/formats/utils/formats_definitions.h                | 5 +++++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc b/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc
index 669b7895..279c840b 100644
--- a/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc
+++ b/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc
@@ -42,7 +42,7 @@ Status TransShapeHwcnToFrazlstm(const DataType &data_type, const std::vector<int
   dst_shape.push_back(src_shape.at(kHwcnN));
   dst_shape.push_back(cube_size);
   dst_shape.push_back(cube_size);
-  if (!CheckShapeValid(dst_shape, kFracZDimsNum)) {
+  if (!CheckShapeValid(dst_shape, kFracZnLstmDimsNum)) {
     GELOGE(ACL_ERROR_GE_SHAPE_INVALID, "Failed to check dst shape %s",
            ShapeToString(dst_shape).c_str());
     return ACL_ERROR_GE_SHAPE_INVALID;
@@ -67,7 +67,7 @@ Status CheckArgsForHwcnToFrazlstm(const TransArgs &args) {
     GELOGE(PARAM_INVALID, "Failed to check src shape %s", ShapeToString(args.src_shape).c_str());
     return PARAM_INVALID;
   }
-  if (!CheckShapeValid(args.dst_shape, kFracZDimsNum)) {
+  if (!CheckShapeValid(args.dst_shape, kFracZnLstmDimsNum)) {
     GELOGE(PARAM_INVALID, "Failed to check dst shape %s", ShapeToString(args.dst_shape).c_str());
     return PARAM_INVALID;
   }
diff --git a/ge/common/formats/utils/formats_definitions.h b/ge/common/formats/utils/formats_definitions.h
index 25f36d6a..01430a9d 100755
--- a/ge/common/formats/utils/formats_definitions.h
+++ b/ge/common/formats/utils/formats_definitions.h
@@ -100,6 +100,11 @@ enum DhwncDimIndex {
   kDhwncC,
   kDhwncDimsNum
 };
+
+enum FracZnLstmIndex {
+  kFracZnLstmDimsNum = 6,
+};
+
 }  // namespace formats
 }  // namespace ge
 #endif  // GE_COMMON_FORMATS_UTILS_FORMATS_DEFINITIONS_H_