Merge branch 'master' into fix/test_celu

kendryte · Aug 28, 2023 · 4835add · 4835add
2 parents 962fc7a + 2cff944
commit 4835add
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/python/nncase/__init__.py b/python/nncase/__init__.py
@@ -357,6 +357,10 @@ class CompileOptions:
     dump_asm: bool
     dump_ir: bool
     dump_dir: str
+    shape_bucket_enable: bool
+    shape_bucket_range_info: dict
+    shape_bucket_segments_count: int
+    shape_bucket_fix_var_map: dict
 
     def __init__(self) -> None:
 
@@ -375,6 +379,10 @@ def __init__(self) -> None:
         self.dump_asm = True
         self.dump_ir = False
         self.dump_dir = "tmp"
+        self.shape_bucket_enable = False
+        self.shape_bucket_range_info = {}
+        self.shape_bucket_segments_count = 2
+        self.shape_bucket_fix_var_map = {}
 
 
 class ShapeBucketOptions: