diff --git a/python/nncase/__init__.py b/python/nncase/__init__.py index 5784a37915..a2376a8717 100644 --- 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: