53 lines
1.1 KiB
C++
53 lines
1.1 KiB
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <frontend_manager/frontend_manager.hpp>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
struct BaseFEParam
|
|
{
|
|
BaseFEParam() {}
|
|
|
|
BaseFEParam(const std::string& name, const std::string& path)
|
|
: m_frontEndName(name)
|
|
, m_modelsPath(path)
|
|
{
|
|
}
|
|
|
|
std::string m_frontEndName;
|
|
std::string m_modelsPath;
|
|
};
|
|
|
|
struct PartShape
|
|
{
|
|
std::string m_modelName;
|
|
std::string m_tensorName;
|
|
ngraph::PartialShape m_oldPartialShape;
|
|
ngraph::PartialShape m_newPartialShape;
|
|
};
|
|
|
|
using PartialShapeParam = std::tuple<BaseFEParam, PartShape>;
|
|
|
|
class FrontEndPartialShapeTest : public ::testing::TestWithParam<PartialShapeParam>
|
|
{
|
|
public:
|
|
BaseFEParam m_baseParam;
|
|
PartShape m_partShape;
|
|
ngraph::frontend::FrontEndManager m_fem;
|
|
ngraph::frontend::FrontEnd::Ptr m_frontEnd;
|
|
ngraph::frontend::InputModel::Ptr m_inputModel;
|
|
|
|
static std::string getTestCaseName(const testing::TestParamInfo<PartialShapeParam>& obj);
|
|
|
|
void SetUp() override;
|
|
|
|
protected:
|
|
void initParamTest();
|
|
|
|
void doLoadFromFile();
|
|
};
|