diff options
Diffstat (limited to 'tests/nnapi/specs/Ex/reduce_sum_ex_4D_float_reducing_C.mod.py')
| -rw-r--r-- | tests/nnapi/specs/Ex/reduce_sum_ex_4D_float_reducing_C.mod.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/nnapi/specs/Ex/reduce_sum_ex_4D_float_reducing_C.mod.py b/tests/nnapi/specs/Ex/reduce_sum_ex_4D_float_reducing_C.mod.py index 4bfea0f97..5d751bd23 100644 --- a/tests/nnapi/specs/Ex/reduce_sum_ex_4D_float_reducing_C.mod.py +++ b/tests/nnapi/specs/Ex/reduce_sum_ex_4D_float_reducing_C.mod.py @@ -17,9 +17,10 @@ model = Model() i1 = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d, %d}" % (batch, rows, cols, depth)) # Axis value should be in the range [-(rank), rank). And '-n' is the same axis with 'rank - n'. So '3' and '-1' are the same axis. axis = Parameter("axis", "TENSOR_INT32", "{2}", [3, -1]) +keepDims = Int32Scalar("keepDims", 0) output = Output("output", "TENSOR_FLOAT32", "{%d, %d, %d}" % (batch, rows, cols)) -model = model.Operation("REDUCE_SUM_EX", i1, axis).To(output) +model = model.Operation("REDUCE_SUM_EX", i1, axis, keepDims).To(output) # Example 1. Input in operand 0, input0 = {i1: # input 0 |