forked from openvinotoolkit/openvino
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcut_specific_model.cpp
31 lines (25 loc) · 1.02 KB
/
cut_specific_model.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
// Copyright (C) 2018-2025 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include "cut_specific_model.hpp"
#include "paddle_utils.hpp"
using namespace ov::frontend;
using PaddleCutTest = FrontEndCutModelTest;
static CutModelParam getTestData_2in_2out() {
CutModelParam res;
res.m_frontEndName = PADDLE_FE;
res.m_modelsPath = std::string(TEST_PADDLE_MODELS_DIRNAME);
res.m_modelName = "2in_2out/2in_2out.pdmodel";
res.m_oldInputs = {"inputX1", "inputX2"};
res.m_newInputs = {"add1.tmp_0"};
res.m_oldOutputs = {"save_infer_model/scale_0.tmp_0", "save_infer_model/scale_1.tmp_0"};
res.m_newOutputs = {"add2.tmp_0"};
res.m_tensorValueName = "conv2dX2.tmp_0";
res.m_tensorValue = {1, 2, 3, 4, 5, 6, 7, 8, 9};
res.m_op_before_name = "conv2dX2.tmp_0";
return res;
}
INSTANTIATE_TEST_SUITE_P(PaddleCutTest,
FrontEndCutModelTest,
::testing::Values(getTestData_2in_2out()),
FrontEndCutModelTest::getTestCaseName);