diff --git a/lite/core/op_lite.cc b/lite/core/op_lite.cc index c3c00d0fa0edeaf3f7070b3e912b9891e44fdc86..dcab292be8f24a6294cb560506f6d03209552d4a 100644 --- a/lite/core/op_lite.cc +++ b/lite/core/op_lite.cc @@ -322,6 +322,7 @@ std::vector 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>(scale_name); }