diff options
Diffstat (limited to 'res/ONNXTests/UNIT_Gemm_000/test.pbtxt')
-rw-r--r-- | res/ONNXTests/UNIT_Gemm_000/test.pbtxt | 79 |
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 } + } + } + } + } +} |