diff options
Diffstat (limited to 'tests/nnapi/specs/V1_0/space_to_depth_float_3.mod.py')
-rw-r--r-- | tests/nnapi/specs/V1_0/space_to_depth_float_3.mod.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/nnapi/specs/V1_0/space_to_depth_float_3.mod.py b/tests/nnapi/specs/V1_0/space_to_depth_float_3.mod.py new file mode 100644 index 000000000..e5298dff1 --- /dev/null +++ b/tests/nnapi/specs/V1_0/space_to_depth_float_3.mod.py @@ -0,0 +1,22 @@ +model = Model() +i1 = Input("input", "TENSOR_FLOAT32", "{1, 4, 4, 2}") +block = Int32Scalar("block_size", 2) +output = Output("output", "TENSOR_FLOAT32", "{1, 2, 2, 8}") + +model = model.Operation("SPACE_TO_DEPTH", i1, block).To(output) + +# Example 1. Input in operand 0, +input0 = {i1: # input 0 + [10, 20, 11, 21, 12, 22, 13, 23, + 14, 24, 15, 25, 16, 26, 17, 27, + 18, 28, 19, 29, 110, 210, 111, 211, + 112, 212, 113, 213, 114, 214, 115, 215]} + +output0 = {output: # output 0 + [10, 20, 11, 21, 14, 24, 15, 25, + 12, 22, 13, 23, 16, 26, 17, 27, + 18, 28, 19, 29, 112, 212, 113, 213, + 110, 210, 111, 211, 114, 214, 115, 215]} + +# Instantiate an example +Example((input0, output0)) |