diff options
-rw-r--r-- | tools/tosa.py | 9 | ||||
-rw-r--r-- | tosa.xml | 4 |
2 files changed, 9 insertions, 4 deletions
diff --git a/tools/tosa.py b/tools/tosa.py index 52fda3e..bc6faa6 100644 --- a/tools/tosa.py +++ b/tools/tosa.py @@ -66,7 +66,7 @@ class TOSASpec: for group in self.xmlroot.findall("./operators/operatorgroup"): self.operatorgroups.append(self.__load_operator_group(group)) for enum in self.xmlroot.findall("./enum"): - self.enums.append(self.load_enum(enum)) + self.enums.append(self.__load_enum(enum)) def __load_version(self): version = self.xmlroot.find("./version") @@ -142,10 +142,15 @@ class TOSASpec: return TOSAOperatorArgument(name, desc, argcats, argtype, shape, levellimits) - def load_enum(self, arg): + def __load_enum(self, arg): name = arg.get("name") desc = arg.get("description").strip() values = [] for val in arg.findall("enumval"): values.append((val.get("name"), val.get("value"), val.get("description"))) return TOSAEnum(name, desc, values) + + def get_enum_by_name(self, name): + for e in self.enums: + if e.name == name: + return e @@ -1977,7 +1977,7 @@ used.</description> <argument category="attribute" name="border" type="int16_t*" shape="[2]"> <description>[border_y, border_x]</description> </argument> - <argument category="attribute" name="mode" type="resize_t" shape="-"> + <argument category="attribute" name="mode" type="resize_mode_t" shape="-"> <description>BILINEAR or NEAREST</description> </argument> <argument category="output" name="output" type="out_t*" shape="[N,OH,OW,C]"> @@ -2284,7 +2284,7 @@ used.</description> </operatorgroup> </operators> - <enum name="resize_t" description="Valid resize types"> + <enum name="resize_mode_t" description="Valid resize types"> <enumval value="0" name="NEAREST_NEIGHBOR" description="Nearest neighbor resize"/> <enumval value="1" name="BILINEAR" description="Bilinear resize"/> </enum> |