IntenseWebs/openvino
3
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Shared onnx tests to API2.0 (#21726)

Browse Source
This commit is contained in:
Vitaliy Urusovskij authored and GitHub committed 2023-12-20 01:03:33 +04:00
1 parent 5b776e9f1f
commit e6ab01c681
2 files changed
+14 -20
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/tests/functional/plugin/shared/include/onnx/quantized_models_tests.hpp
+4 -4
View File
@@ -5,19 +5,19 @@
#pragma once
#include <string>
#include "shared_test_classes/base/layer_test_utils.hpp"
#include "shared_test_classes/base/ov_subgraph.hpp"
namespace ONNXTestsDefinitions {
class QuantizedModelsTests : public testing::WithParamInterface<std::string>,
virtual public LayerTestsUtils::LayerTestsCommon {
virtual public ov::test::SubgraphBaseStaticTest {
public:
static std::string getTestCaseName(const testing::TestParamInfo<std::string>& obj);
protected:
void SetUp() override;
using LayerInputTypes = std::unordered_map<std::string, std::vector<ngraph::element::Type_t>>;
void runModel(const char* model, const LayerInputTypes& expected_layer_input_types, float thr);
using LayerInputTypes = std::unordered_map<std::string, std::vector<ov::element::Type>>;
void run_model(const char* model, const LayerInputTypes& expected_layer_input_types, float thr);
};
} // namespace ONNXTestsDefinitions
src/tests/functional/plugin/shared/src/onnx/quantized_models_tests.cpp
+10 -16
View File
@@ -24,15 +24,13 @@ static std::string getModelFullPath(const char* path) {
FileUtils::makePath<char>(ov::test::utils::getExecutableDirectory(), TEST_MODELS), path);
}
void QuantizedModelsTests::runModel(const char* model, const LayerInputTypes& expected_layer_input_types, float thr) {
threshold = thr;
auto ie = getCore();
auto network = ie->ReadNetwork(getModelFullPath(model));
function = network.getFunction();
Run();
auto runtime_function = executableNetwork.GetExecGraphInfo().getFunction();
void QuantizedModelsTests::run_model(const char* model, const LayerInputTypes& expected_layer_input_types, float thr) {
abs_threshold = thr;
function = core->read_model(getModelFullPath(model));
ov::test::SubgraphBaseStaticTest::run();
auto runtime_model = compiledModel.get_runtime_model();
int ops_found = 0;
for (const auto& node : runtime_function->get_ordered_ops()) {
for (const auto& node : runtime_model->get_ordered_ops()) {
const auto& name = node->get_friendly_name();
if (expected_layer_input_types.count(name)) {
ops_found++;
@@ -47,25 +45,21 @@ void QuantizedModelsTests::runModel(const char* model, const LayerInputTypes& ex
}
TEST_P(QuantizedModelsTests, MaxPoolQDQ) {
SKIP_IF_CURRENT_TEST_IS_DISABLED();
runModel("max_pool_qdq.onnx", {{"890_original", {ngraph::element::u8}}}, 1e-5);
run_model("max_pool_qdq.onnx", {{"890_original", {ov::element::u8}}}, 1e-5);
}
TEST_P(QuantizedModelsTests, MaxPoolFQ) {
SKIP_IF_CURRENT_TEST_IS_DISABLED();
runModel("max_pool_fq.onnx", {{"887_original", {ngraph::element::u8}}}, 1e-5);
run_model("max_pool_fq.onnx", {{"887_original", {ov::element::u8}}}, 1e-5);
}
TEST_P(QuantizedModelsTests, ConvolutionQDQ) {
SKIP_IF_CURRENT_TEST_IS_DISABLED();
// activations have type uint8 and weights int8
runModel("convolution_qdq.onnx", {{"908_original", {ngraph::element::u8, ngraph::element::i8}}}, 1.5e-2);
run_model("convolution_qdq.onnx", {{"908_original", {ov::element::u8, ov::element::i8}}}, 1.5e-2);
}
TEST_P(QuantizedModelsTests, ConvolutionFQ) {
SKIP_IF_CURRENT_TEST_IS_DISABLED();
// activations have type uint8 and weights int8
runModel("convolution_fq.onnx", {{"902_original", {ngraph::element::u8, ngraph::element::i8}}}, 1.5e-2);
run_model("convolution_fq.onnx", {{"902_original", {ov::element::u8, ov::element::i8}}}, 1.5e-2);
}
} // namespace ONNXTestsDefinitions