diff options
| author | TatWai Chong <tatwai.chong@arm.com> | 2022-11-11 15:28:50 -0800 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-03-02 00:37:14 +0000 |
| commit | 6ad6d6e6c4236317c0835f93b0ff2647e628c29e (patch) | |
| tree | 23298bf6d8d45e910ca310d2a40287cd2bacf3b9 | |
| parent | 0df72e9a6dd2b02174a0e203c37f5d5ef89a5b57 (diff) | |
| download | reference_model-6ad6d6e6c4236317c0835f93b0ff2647e628c29e.tar.gz | |
Extend reduce_all to test with tflite
Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: Ifb3ed9c448005d6d53c5d0938b5146003a26e8a3
| -rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 931aef6..c11a83d 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -208,7 +208,7 @@ TF_OP_LIST = { "reduce_all": { "operands": (1, 0), "build_fcn": (TBuilder.ReduceAll, TGen.tgBasic, ArgGen.agAxesListKeepdims), - "types": {"tf": TYPE_B}, + "types": TYPE_B, }, "reduce_min": { "operands": (1, 0), |