summaryrefslogtreecommitdiff
path: root/externals/nnapi_test_generator/tests/P_quantized_conv/quantized.mod.py
diff options
context:
space:
mode:
Diffstat (limited to 'externals/nnapi_test_generator/tests/P_quantized_conv/quantized.mod.py')
-rw-r--r--externals/nnapi_test_generator/tests/P_quantized_conv/quantized.mod.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/externals/nnapi_test_generator/tests/P_quantized_conv/quantized.mod.py b/externals/nnapi_test_generator/tests/P_quantized_conv/quantized.mod.py
new file mode 100644
index 000000000..7ef623513
--- /dev/null
+++ b/externals/nnapi_test_generator/tests/P_quantized_conv/quantized.mod.py
@@ -0,0 +1,11 @@
+i4 = Int32Scalar("b4", 2)
+i5 = Int32Scalar("b5", 2)
+i6 = Int32Scalar("b6", 2)
+i7 = Int32Scalar("b7", 0)
+i2 = Input("op2", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}") # input 0
+i3 = Output("op3", "TENSOR_QUANT8_ASYMM", "{1, 1, 1, 1}") # output 0
+i0 = Parameter("op0", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}", [1, 1, 1, 1]) # parameters
+i1 = Parameter("op1", "TENSOR_INT32", "{1}", [0]) # parameters
+model = Model()
+model = model.Conv(i2, i0, i1, i4, i5, i6, i7).To(i3)
+