Commit 9b3d197a authored by suryasidd's avatar suryasidd Committed by Scott Cyphers

Matmul float type test case for UEP (#3877)

* Matmul float type test case for UEP
Signed-off-by: 's avatarsuryasidd <surya.siddharth.pemmaraju@intel.com>

* Removed microsoft ops domains and ran clang-format
Signed-off-by: 's avatarsuryasidd <surya.siddharth.pemmaraju@intel.com>
parent d317bad5
ir_version: 6
graph {
node {
input: "A"
input: "B"
output: "Y"
name: "node1"
op_type: "MatMul"
doc_string: "MatMul"
}
name: "fused_function_subgraph"
doc_string: ""
input {
name: "A"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 3
}
dim {
dim_value: 1
}
dim {
dim_value: 2
}
}
}
}
}
input {
name: "B"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 2
}
}
}
}
}
output {
name: "Y"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 3
}
dim {
dim_value: 1
}
}
}
}
}
}
opset_import {
domain: ""
version: 7
}
opset_import {
domain: "ai.onnx.ml"
version: 2
}
......@@ -1703,3 +1703,14 @@ NGRAPH_TEST(onnx_${BACKEND_NAME}, model_reverse_sequence_time_and_batch_axis_equ
ngraph_error)
<< "ReverseSequence 'time_axis' and 'batch_axis' can't be equal.";
}
NGRAPH_TEST(onnx_${BACKEND_NAME}, matmul_float_type)
{
auto function = onnx_import::import_onnx_model(
file_util::path_join(SERIALIZED_ZOO, "onnx/matmul_float.prototxt"));
auto test_case = ngraph::test::NgraphTestCase(function, "${BACKEND_NAME}");
test_case.add_input<float>(std::vector<float>{0, 1, 2, 3, 4, 5});
test_case.add_input<float>(std::vector<float>{0, 1});
test_case.add_expected_output<float>(Shape{3, 1}, std::vector<float>{1, 3, 5});
test_case.run();
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment