diff options
Diffstat (limited to 'verif/generator/tosa_utils.py')
-rw-r--r-- | verif/generator/tosa_utils.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index 8ff62f1..f9df8d5 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -18,6 +18,7 @@ DTYPE_ATTRIBUTES = { DType.UINT16: {"str": "u16", "width": 16}, DType.INT32: {"str": "i32", "width": 32}, DType.INT48: {"str": "i48", "width": 48}, + DType.SHAPE: {"str": "i64", "width": 64}, DType.FP16: {"str": "f16", "width": 16}, DType.BF16: {"str": "bf16", "width": 16}, DType.FP32: {"str": "f32", "width": 32}, |