165db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wangmodel = Model()
265db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wangi1 = Input("input", "TENSOR_QUANT8_ASYMM", "{4, 2, 2, 1}, 1.0, 0")
365db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wangblock = Parameter("block_size", "TENSOR_INT32", "{2}", [2, 2])
465db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wangoutput = Output("output", "TENSOR_QUANT8_ASYMM", "{1, 4, 4, 1}, 1.0, 0")
565db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang
6a9ce0001ca1df32fde3f322c385eef8ce7279329Miao Wangmodel = model.Operation("BATCH_TO_SPACE_ND", i1, block).To(output)
765db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang
865db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang# Example 1. Input in operand 0,
965db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wanginput0 = {i1: # input 0
1065db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang          [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]}
1165db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang
1265db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wangoutput0 = {output: # output 0
1365db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang           [1, 5, 2, 6, 9, 13, 10, 14, 3, 7, 4, 8, 11, 15, 12, 16]}
1465db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang
1565db490301bb2e35864fb0f2aef67fa8c4be2b88Miao Wang# Instantiate an example
1665db490301bb2e35864fb0f2aef67fa8c4be2b88Miao WangExample((input0, output0))