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

   1 # model
   2 input0 = Input("input0", "TENSOR_FLOAT32", "{12}")
   3 size_splits = Input("size_splits", "TENSOR_INT32", "{3}")
   4 split_dim = Input("split_dim", "TENSOR_INT32", "{1}")
   5 num_splits = Int32Scalar("num_splits",3);
   6
   7 output0 = Output("output0", "TENSOR_FLOAT32", "{3}")
   8 output1 = Output("output1", "TENSOR_FLOAT32", "{5}")
   9 output2 = Output("output2", "TENSOR_FLOAT32", "{4}")
  10
  11 model = Model().Operation("SPLIT_V_EX", input0, size_splits, split_dim, num_splits).To((output0, output1, output2))
  12
  13 # Example 1.
  14 input_dict = {
  15     input0: [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0],
  16     size_splits: [3, 5, 4],
  17     split_dim: [0]
  18 }
  19 output_dict = {
  20     output0: [1.0, 2.0, 3.0],
  21     output1: [4.0, 5.0, 6.0, 7.0, 8.0],
  22     output2: [9.0, 10.0, 11.0, 12.0]
  23 }
  24
  25 Example((input_dict, output_dict))