diff options
Diffstat (limited to 'tests/nnapi/specs/V1_2/topk_v2_1D_float_nnfw.mod.py')
| -rwxr-xr-x[-rw-r--r--] | tests/nnapi/specs/V1_2/topk_v2_1D_float_nnfw.mod.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/nnapi/specs/V1_2/topk_v2_1D_float_nnfw.mod.py b/tests/nnapi/specs/V1_2/topk_v2_1D_float_nnfw.mod.py index 1e0ed21d6..5be9c49bf 100644..100755 --- a/tests/nnapi/specs/V1_2/topk_v2_1D_float_nnfw.mod.py +++ b/tests/nnapi/specs/V1_2/topk_v2_1D_float_nnfw.mod.py @@ -4,7 +4,7 @@ i1 = Input("op1", "TENSOR_FLOAT32", "{4}") # a vector of input k = Int32Scalar("k", 2) i2 = Output("op2", "TENSOR_FLOAT32", "{2}") # values of output i3 = Output("op3", "TENSOR_INT32", "{2}") # indexes of output -model = model.Operation("TOPK_V2_EX", i1, k).To([i2, i3]) +model = model.Operation("TOPK_V2", i1, k).To([i2, i3]) # Example 1. Input in operand 0, input0 = {i1: # input 0 |