diff options
| author | Les Bell <les.bell@arm.com> | 2021-08-10 08:34:43 +0100 |
|---|---|---|
| committer | Les Bell <les.bell@arm.com> | 2021-08-10 10:12:55 +0100 |
| commit | 33d837e3235d709596cd10d43db18399281b436a (patch) | |
| tree | 93f9c5b5a92a6a559680967049440a0111f1e133 | |
| parent | 30e4680b789c728d503cf76ee316468969943d51 (diff) | |
| download | reference_model-33d837e3235d709596cd10d43db18399281b436a.tar.gz | |
fix output type for resize in test gen
Signed-off-by: Les Bell <les.bell@arm.com>
Change-Id: I9daf8fd691af74500819718519caac30c966a1f2
| -rw-r--r-- | verif/tosa_test_gen.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index c458538..99dc5f8 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -733,11 +733,11 @@ class TosaArgGen: # Exclude illegal {mode, type} configurations. Pick legal output types if m == ResizeMode.NEAREST and dtype == DType.INT8: - outputDTypeList = [DType.INT32] + outputDTypeList = [DType.INT8] elif m == ResizeMode.NEAREST and dtype == DType.INT16: outputDTypeList = [DType.INT16] elif m == ResizeMode.BILINEAR and dtype == DType.INT8: - outputDTypeList = [DType.INT8] + outputDTypeList = [DType.INT32] elif m == ResizeMode.BILINEAR and dtype == DType.INT16: outputDTypeList = [DType.INT48] elif dtype == DType.FLOAT: @@ -770,7 +770,7 @@ class TosaArgGen: arg_list.append( ( "mode{}_odim{}x{}_out{}_st{:.2f}x{:.2f}_off{:.2f}x{:.2f}".format( - m, + "N" if m == ResizeMode.NEAREST else "B", output_dims[0], output_dims[1], testGen.typeStr(outputDType), @@ -824,7 +824,7 @@ class TosaArgGen: arg_list.append( ( "mode{}_shift{}_odim{}x{}_out{}_st{}x{}_off{}x{}".format( - m, + "N" if m == ResizeMode.NEAREST else "B", shift, output_dims[0], output_dims[1], |