diff options
author | TatWai Chong <tatwai.chong@arm.com> | 2023-02-22 18:38:01 -0800 |
---|---|---|
committer | TatWai Chong <tatwai.chong@arm.com> | 2023-02-24 14:54:16 -0800 |
commit | 2226f90d5a6c48a975045bc9e0419113ce764aaf (patch) | |
tree | 78903dabce38db04e938414919fd45142fc72f4f /verif/frameworks/tosa_verif_framework_generator.py | |
parent | 681e0a8085d19629eddfb3d76a2ef9c0a7110f3c (diff) | |
download | reference_model-2226f90d5a6c48a975045bc9e0419113ce764aaf.tar.gz |
Extend pad tests to include quantized type and explicit pad value.
Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: I4a49f45aa73044aff5b0a8b3dba58c1f52c1ae21
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_generator.py')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 60106a1..931aef6 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -631,7 +631,10 @@ TF_OP_LIST = { "pad": { "operands": (1, 0), "build_fcn": (TBuilder.Pad, TGen.tgBasic, ArgGen.agPad), - "types": TYPE_F, + "types": { + "tf": TYPE_F, + "tflite": list(TYPE_F + [QuantType.ALL_U8, QuantType.ALL_I8]), + }, }, "expand_dims": { "operands": (1, 0), |