From 0be4ef9f59315c8ce482f6880fc7c856886d9ecf Mon Sep 17 00:00:00 2001
From: hong19860320 <9973393+hong19860320@users.noreply.github.com>
Date: Tue, 15 Sep 2020 20:49:45 +0800
Subject: [PATCH] [Core] Fix the missing of the attr name of the output scale
 (#4334)

---
 lite/core/op_lite.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lite/core/op_lite.cc b/lite/core/op_lite.cc
index c3c00d0fa0..dcab292be8 100644
--- a/lite/core/op_lite.cc
+++ b/lite/core/op_lite.cc
@@ -322,6 +322,7 @@ std::vector<float> OpInfo::GetOutputScale(const std::string &name,
     int index;
     CHECK(GetOutputArgname(name, &argname));
     CHECK(GetOutputIndex(name, &index));
+    scale_name = argname + to_string(index) + "_scale";
   }
   return GetAttr<std::vector<float>>(scale_name);
 }
-- 
GitLab