diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-09-27 13:50:00 +0100 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-10-13 18:21:15 +0100 |
| commit | bc2a3db54ecee48fe2236f7fc03da8fd07d81ca0 (patch) | |
| tree | c3908f23c369fd3226e840f81c3ba4b49cc409a0 /verif/tests/test_tosa_refmodel.py | |
| parent | 93d4390f9aa5c4369f889e1cd336aa4e809ff6a7 (diff) | |
| download | reference_model-bc2a3db54ecee48fe2236f7fc03da8fd07d81ca0.tar.gz | |
Rename FLOAT type to FP32
Update tensor operations naming to state input type as TxT in
all cases. Effects CONV2D, CONV3D, DEPTHWISE_CONV2D,
FULLY_CONNECTED, TRANSPOSE_CONV2D.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ic959acfcb3aa0a910b33b774a5a85fac08219205
Diffstat (limited to 'verif/tests/test_tosa_refmodel.py')
| -rw-r--r-- | verif/tests/test_tosa_refmodel.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/verif/tests/test_tosa_refmodel.py b/verif/tests/test_tosa_refmodel.py index fbe3a7d..b608fd8 100644 --- a/verif/tests/test_tosa_refmodel.py +++ b/verif/tests/test_tosa_refmodel.py @@ -45,7 +45,7 @@ REF_MODEL_TYPE_TO_OUT = { "uint8": "u8", "int16": "i16", "int32": "i32", - "float": "float", + "fp32": "f32", "fp16": "f16", } @@ -123,21 +123,21 @@ class BuildTosaTest: # Tests - op_name, ref_model_type, num_expected_tests TEST_PARAMS = [ ("add", "int32", 1), - ("add", "float", 1), + ("add", "fp32", 1), ("abs", "int32", 1), - ("abs", "float", 1), + ("abs", "fp32", 1), ("abs", "fp16", 1), ("negate", "int8", 1), ("negate", "int16", 1), ("negate", "int32", 1), - ("negate", "float", 1), + ("negate", "fp32", 1), ("negate", "fp16", 1), # One test per axis (shape dimensions) ("concat", "bool", SHAPE_DIMS), ("concat", "int8", SHAPE_DIMS), ("concat", "int16", SHAPE_DIMS), ("concat", "int32", SHAPE_DIMS), - ("concat", "float", SHAPE_DIMS), + ("concat", "fp32", SHAPE_DIMS), ("concat", "fp16", SHAPE_DIMS), ] |