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

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