Add shape_bucket attributes into CompileOptions.

kendryte · Aug 25, 2023 · a7a5a0c · a7a5a0c
1 parent d873ae9
commit a7a5a0c
Showing 1 changed file with 4 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: