Skip to content

Commit

Permalink
(feat): add support for microsoft.range operator for ONNX frontend
Browse files Browse the repository at this point in the history
Signed-off-by: 11happy <[email protected]>
  • Loading branch information
11happy committed Dec 23, 2024
1 parent d156895 commit 6754c25
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions src/frontends/onnx/tests/onnx_import_com_microsoft.in.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1457,6 +1457,30 @@ OPENVINO_TEST(${BACKEND_NAME}, onnx_com_microsoft_fusedmatmul_2x5x3x6x4_2x6x3x4x
test_case.run();
}

OPENVINO_TEST(${BACKEND_NAME}, onnx_com_microsoft_range_with_delta) {
const auto model = convert_model("com.microsoft/range_with_delta.onnx");
auto test_case = ov::test::TestCase(model, s_device);

test_case.add_input<float>({0.f});
test_case.add_input<float>({10.f});
test_case.add_input<float>({1.f});
test_case.add_expected_output<float>(Shape{10},
{0.f,1.f,2.f,3.f,4.f,5.f,6.f,7.f,8.f,9.f});

test_case.run();
}

OPENVINO_TEST(${BACKEND_NAME}, onnx_com_microsoft_range_without_delta) {
const auto model = convert_model("com.microsoft/range_without_delta.onnx");
auto test_case = ov::test::TestCase(model, s_device);

test_case.add_input<float>({0.f});
test_case.add_input<float>({10.f});
test_case.add_expected_output<float>(Shape{10},
{0.f,1.f,2.f,3.f,4.f,5.f,6.f,7.f,8.f,9.f});

test_case.run();
}
OPENVINO_TEST(${BACKEND_NAME}, onnx_com_microsoft_matmul_integer_to_float) {
const auto model = convert_model("com.microsoft/matmul_integer_to_float.onnx");
auto test_case = ov::test::TestCase(model, s_device);
Expand Down

0 comments on commit 6754c25

Please sign in to comment.