Imported Upstream version 1.7.0
[platform/core/ml/nnfw.git] / tests / nnapi / specs / V1_2 / reduce_max_quant8_1_nnfw.mod.py
1 model = Model()
2 i1 = Input("input", "TENSOR_QUANT8_ASYMM", "{4, 3, 2}, 0.8, 5")
3 axis = Parameter("axis", "TENSOR_INT32", "{4}", [1, 0, -3, -3])
4 keepDims = False
5 output = Output("output", "TENSOR_QUANT8_ASYMM", "{2}, 0.8, 5")
6
7 model = model.Operation("REDUCE_MAX", i1, axis, keepDims).To(output)
8
9 # Example 1. Input in operand 0,
10 input0 = {i1: # input 0
11           [1,  2,  3,  4,  5,  6,  7,  8,
12            9,  10, 11, 12, 13, 14, 15, 16,
13            17, 18, 19, 20, 21, 22, 23, 24]}
14
15 output0 = {output: # output 0
16           [23, 24]}
17
18 # Instantiate an example
19 Example((input0, output0))