tests/nnapi/specs/Ex/split_v_ex_1D_float_1_nnfw.mod.py

   1 # model
   2 model = Model()
   3 i1 = Input("op1", "TENSOR_FLOAT32", "{8}")
   4 size_splits = Input("size_splits", "TENSOR_INT32", "{8}")
   5 split_dim = Input("split_dim", "TENSOR_INT32", "{1}")
   6 num_splits = Int32Scalar("num_splits", 8)
   7
   8 i2 = Output("op2", "TENSOR_FLOAT32", "{1}")
   9 i3 = Output("op3", "TENSOR_FLOAT32", "{1}")
  10 i4 = Output("op4", "TENSOR_FLOAT32", "{1}")
  11 i5 = Output("op5", "TENSOR_FLOAT32", "{1}")
  12 i6 = Output("op6", "TENSOR_FLOAT32", "{1}")
  13 i7 = Output("op7", "TENSOR_FLOAT32", "{1}")
  14 i8 = Output("op8", "TENSOR_FLOAT32", "{1}")
  15 i9 = Output("op9", "TENSOR_FLOAT32", "{1}")
  16
  17 model = model.Operation("SPLIT_V_EX", i1, size_splits, split_dim, num_splits).To([i2, i3, i4, i5, i6, i7, i8, i9])
  18
  19 # Example 1. Input in operand 0,
  20 input0 = {i1: # input 0
  21           [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0],
  22           size_splits:
  23           [1, 1, 1, 1, 1, 1, 1, 1],
  24           split_dim:
  25           [0]
  26           }
  27
  28 output0 = {
  29     i2: # output 0
  30           [1.0],
  31     i3: # output 1
  32           [2.0],
  33     i4: # output 2
  34           [3.0],
  35     i5: # output 3
  36           [4.0],
  37     i6: # output 4
  38           [5.0],
  39     i7: # output 5
  40           [6.0],
  41     i8: # output 6
  42           [7.0],
  43     i9: # output 7
  44           [8.0]}
  45
  46 # Instantiate an example
  47 Example((input0, output0))