tests/nnapi/specs/V1_2/reduce_max_2D_int32_nnfw.mod.py

   1 # model
   2 model = Model()
   3 i1 = Input("input", "TENSOR_INT32", "{3, 4}")
   4 axis = Int32Scalar("axis", 1)
   5 keepDims = False
   6 out1 = Output("output", "TENSOR_INT32", "{3}")
   7 model = model.Operation("REDUCE_MAX", i1, axis, keepDims).To(out1)
   8
   9 # Example 1. Input in operand 0, 1
  10 input0 = {i1: # input 0
  11           [3, 11, 3, 5,
  12            28, 0, -1, -13,
  13            -4, -22, -2, -49]}
  14
  15 output0 = {out1: # output 0
  16            [11, 28, -2]}
  17
  18 # Instantiate an example
  19 Example((input0, output0))