diff options
| author | Won Jeon <won.jeon@arm.com> | 2023-08-10 10:33:01 +0000 |
|---|---|---|
| committer | Won Jeon <won.jeon@arm.com> | 2023-08-18 15:21:15 -0700 |
| commit | a21b2e88d19d8cb11a9120d40bacbb594d600565 (patch) | |
| tree | 3bc8a40db72a31c1e552a3bd6339627a1175686e /verif/generator/tosa_utils.py | |
| parent | e0247481eb1f83f6eb7161d3f7ac2690b180952a (diff) | |
| download | reference_model-a21b2e88d19d8cb11a9120d40bacbb594d600565.tar.gz | |
Add DIM operator to reference model
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iea11ee5d3d98773e9c5e9b827593c05afb41ce3b
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}, |