3 i1 = Input("op1", "TENSOR_INT32", "{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)
8 i2 = Output("op2", "TENSOR_INT32", "{1}")
9 i3 = Output("op3", "TENSOR_INT32", "{1}")
10 i4 = Output("op4", "TENSOR_INT32", "{1}")
11 i5 = Output("op5", "TENSOR_INT32", "{1}")
12 i6 = Output("op6", "TENSOR_INT32", "{1}")
13 i7 = Output("op7", "TENSOR_INT32", "{1}")
14 i8 = Output("op8", "TENSOR_INT32", "{1}")
15 i9 = Output("op9", "TENSOR_INT32", "{1}")
17 model = model.Operation("SPLIT_V_EX", i1, size_splits, split_dim, num_splits).To([i2, i3, i4, i5, i6, i7, i8, i9])
19 # Example 1. Input in operand 0,
20 input0 = {i1: # input 0
21 [1, 2, 3, 4, 5, 6, 7, 8],
23 [1, 1, 1, 1, 1, 1, 1, 1],
46 # Instantiate an example
47 Example((input0, output0))