summaryrefslogtreecommitdiff
path: root/res/ONNXTests/UNIT_Gemm_000/test.pbtxt
diff options
context:
space:
mode:
Diffstat (limited to 'res/ONNXTests/UNIT_Gemm_000/test.pbtxt')
-rw-r--r--res/ONNXTests/UNIT_Gemm_000/test.pbtxt79
1 files changed, 79 insertions, 0 deletions
diff --git a/res/ONNXTests/UNIT_Gemm_000/test.pbtxt b/res/ONNXTests/UNIT_Gemm_000/test.pbtxt
new file mode 100644
index 000000000..6fd497f89
--- /dev/null
+++ b/res/ONNXTests/UNIT_Gemm_000/test.pbtxt
@@ -0,0 +1,79 @@
+# This testcase is compatible with ONNX 1.4.1 or newer
+ir_version: 5
+
+opset_import {
+ version: 7
+}
+
+graph {
+ name: "Gemm_000"
+
+ node {
+ input: "input"
+ input: "weight"
+ input: "bias"
+ output: "output"
+ op_type: "Gemm"
+ attribute {
+ name: "alpha"
+ f: 1.5
+ type: FLOAT
+ }
+ attribute {
+ name: "beta"
+ f: 1.5
+ type: FLOAT
+ }
+ attribute {
+ name: "transA"
+ i: 1
+ type: INT
+ }
+ }
+
+# Initializers generated by python helper script:
+# a = np.ones((2,2), dtype = np.float32)
+# onnx.numpy_helper.from_array(a)
+#
+ initializer {
+ dims: 3
+ dims: 2
+ data_type: 1
+ name: "weight"
+ raw_data: "\000\000\200?\000\000\200?\000\000\200?\000\000\200?\000\000\200?\000\000\200?"
+ }
+
+ initializer {
+ dims: 2
+ dims: 2
+ data_type: 1
+ name: "bias"
+ raw_data: "\000\000\200?\000\000\200?\000\000\200?\000\000\200?"
+ }
+
+ input {
+ name: "input"
+ type {
+ tensor_type {
+ elem_type: 1 # FLOAT type
+ shape {
+ dim { dim_value: 3 }
+ dim { dim_value: 2 }
+ }
+ }
+ }
+ }
+
+ output {
+ name: "output"
+ type {
+ tensor_type {
+ elem_type: 1 # FLOAT type
+ shape {
+ dim { dim_value: 2 }
+ dim { dim_value: 2 }
+ }
+ }
+ }
+ }
+}