diff options
Diffstat (limited to 'tests/nnapi/specs/V1_2/topk_v2_2D_float_nnfw.mod.py')
| -rwxr-xr-x[-rw-r--r--] | tests/nnapi/specs/V1_2/topk_v2_2D_float_nnfw.mod.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/nnapi/specs/V1_2/topk_v2_2D_float_nnfw.mod.py b/tests/nnapi/specs/V1_2/topk_v2_2D_float_nnfw.mod.py index 204bc143f..29113b901 100644..100755 --- a/tests/nnapi/specs/V1_2/topk_v2_2D_float_nnfw.mod.py +++ b/tests/nnapi/specs/V1_2/topk_v2_2D_float_nnfw.mod.py @@ -4,7 +4,7 @@ i1 = Input("op1", "TENSOR_FLOAT32", "{3,4}") # a matirx of input k = Int32Scalar("k", 2) o1 = Output("op2", "TENSOR_FLOAT32", "{3,2}") # values of output o2 = Output("op3", "TENSOR_INT32", "{3,2}") # indexes of output -model = model.Operation("TOPK_V2_EX", i1, k).To([o1, o2]) +model = model.Operation("TOPK_V2", i1, k).To([o1, o2]) # Example 1. Input in operand 0, input0 = {i1: # input 0 @@ -13,7 +13,7 @@ input0 = {i1: # input 0 2.123456789123456789, 18.123456789123456789, 19.123456789123456789, 11.123456789123456789]} output0 = {o1: # output 1 - [6.123456789123456789, 5.123456789123456789, + [6.123456789123456789, 5.123456789123456789, 9.123456789123456789, 8.123456789123456789, 19.123456789123456789, 18.123456789123456789], o2: # output 1 |