Add gtests to test Pad op for 2D/3D tensors (#3946)
authorShubham Gupta/SNAP /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com>
Tue, 18 Dec 2018 01:08:00 +0000 (06:38 +0530)
committer오형석/On-Device Lab./Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Tue, 18 Dec 2018 01:08:00 +0000 (10:08 +0900)
This patch will add Pad gtest to test for 2D/3D tensors

Signed-off-by: shubham <shub98.gupta@samsung.com>
runtimes/tests/neural_networks_test/specs/V1_1/pad_2D_HW_nnfw.mod.py [new file with mode: 0644]
runtimes/tests/neural_networks_test/specs/V1_1/pad_3D_HWC_nnfw.mod.py [new file with mode: 0644]

diff --git a/runtimes/tests/neural_networks_test/specs/V1_1/pad_2D_HW_nnfw.mod.py b/runtimes/tests/neural_networks_test/specs/V1_1/pad_2D_HW_nnfw.mod.py
new file mode 100644 (file)
index 0000000..d21a4f1
--- /dev/null
@@ -0,0 +1,21 @@
+# model
+model = Model()
+i1 = Input("op1", "TENSOR_FLOAT32", "{2, 2}")
+i2 = Parameter("op2", "TENSOR_INT32", "{2, 2}", [1, 2, 2, 1])
+i3 = Output("op3", "TENSOR_FLOAT32", "{5, 5}")
+model = model.Operation("PAD", i1, i2).To(i3)
+
+# Example 1. Input in operand 0,
+input0 = {i1: # input 0
+          [1.0, 2.0,
+           3.0, 4.0,]}
+
+output0 = {i3: # output 0
+           [0.0, 0.0, 0.0, 0.0, 0.0,
+           0.0, 0.0, 1.0, 2.0, 0.0,
+            0.0, 0.0, 3.0, 4.0, 0.0,
+            0.0, 0.0, 0.0, 0.0, 0.0,
+           0.0, 0.0, 0.0, 0.0, 0.0]}
+
+# Instantiate an example
+Example((input0, output0))
diff --git a/runtimes/tests/neural_networks_test/specs/V1_1/pad_3D_HWC_nnfw.mod.py b/runtimes/tests/neural_networks_test/specs/V1_1/pad_3D_HWC_nnfw.mod.py
new file mode 100644 (file)
index 0000000..d98ec6c
--- /dev/null
@@ -0,0 +1,27 @@
+# model
+model = Model()
+i1 = Input("op1", "TENSOR_FLOAT32", "{2, 2, 2}")
+i2 = Parameter("op2", "TENSOR_INT32", "{3, 2}", [1, 0, 0, 1, 1, 1])
+i3 = Output("op3", "TENSOR_FLOAT32", "{3, 3, 4}")
+model = model.Operation("PAD", i1, i2).To(i3)
+
+# Example 1. Input in operand 0,
+input0 = {i1: # input 0
+          [1.0, 2.0,
+          3.0, 4.0,
+          5.0, 6.0,
+          7.0, 8.0]}
+
+output0 = {i3: # output 0
+           [0.0, 0.0, 0.0, 0.0,
+            0.0, 0.0, 0.0, 0.0,
+           0.0, 0.0, 0.0, 0.0,
+           0.0, 1.0, 2.0, 0.0,
+           0.0, 3.0, 4.0, 0.0,
+           0.0, 0.0, 0.0, 0.0,
+           0.0, 5.0, 6.0, 0.0,
+           0.0, 7.0, 8.0, 0.0,
+           0.0, 0.0, 0.0, 0.0]}
+
+# Instantiate an example
+Example((input0, output0))