class LogicalLayerTestVPU : public LogicalLayerTest {
protected:
void SetUp() override {
- const auto& inputShapes = std::get<0>(GetParam());
- const auto& ngInputsPrecision = std::get<4>(GetParam());
- const auto& logicalOpType = std::get<1>(GetParam());
- targetDevice = std::get<5>(GetParam());
- const auto& additionalConfig = std::get<9>(GetParam());
- configuration.insert(additionalConfig.begin(), additionalConfig.end());
- outPrc = ngInputsPrecision;
-
- auto ngInputsPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(ngInputsPrecision);
+ SetupParams();
+
+ auto ngInputsPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(inPrc);
auto inputs = ngraph::builder::makeParams(ngInputsPrc, {inputShapes.first, logicalOpType != ngraph::helpers::LogicalTypes::LOGICAL_NOT ?
inputShapes.second : ngraph::Shape()});
ngraph::NodeVector convertedInputs;
::testing::Values(ngraph::helpers::InputLayerType::PARAMETER),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::I32),
- ::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
+ ::testing::Values(InferenceEngine::Precision::I32),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(CommonTestUtils::DEVICE_MYRIAD),
::testing::Values(ngraph::helpers::InputLayerType::CONSTANT),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::I32),
- ::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
+ ::testing::Values(InferenceEngine::Precision::I32),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(CommonTestUtils::DEVICE_MYRIAD),
virtual public LayerTestsUtils::LayerTestsCommon {
protected:
InferenceEngine::Blob::Ptr GenerateInput(const InferenceEngine::InputInfo &info) const override;
+ void SetupParams();
void SetUp() override;
public:
static std::string getTestCaseName(testing::TestParamInfo<LogicalTestParams> obj);
static std::vector<LogicalParams::InputShapesTuple> combineShapes(const std::map<std::vector<size_t>, std::vector<std::vector<size_t >>>& inputShapes);
+
+protected:
+ LogicalParams::InputShapesTuple inputShapes;
+ ngraph::helpers::LogicalTypes logicalOpType;
+ ngraph::helpers::InputLayerType secondInputType;
+ InferenceEngine::Precision netPrecision;
+ std::map<std::string, std::string> additional_config;
};
} // namespace LayerTestsDefinitions
return FuncTestUtils::createAndFillBlob(info.getTensorDesc(), 2, 0);
}
-void LogicalLayerTest::SetUp() {
- InputShapesTuple inputShapes;
- InferenceEngine::Precision inputsPrecision;
- ngraph::helpers::LogicalTypes logicalOpType;
- ngraph::helpers::InputLayerType secondInputType;
- InferenceEngine::Precision netPrecision;
- std::string targetName;
- std::map<std::string, std::string> additional_config;
- std::tie(inputShapes, logicalOpType, secondInputType, netPrecision, inPrc, outPrc, inLayout, outLayout, targetDevice, additional_config) =
+void LogicalLayerTest::SetupParams() {
+ std::tie(inputShapes, logicalOpType, secondInputType, netPrecision,
+ inPrc, outPrc, inLayout, outLayout, targetDevice, additional_config) =
this->GetParam();
- auto ngInputsPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(inPrc);
configuration.insert(additional_config.begin(), additional_config.end());
+}
+void LogicalLayerTest::SetUp() {
+ SetupParams();
+
+ auto ngInputsPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(inPrc);
auto inputs = ngraph::builder::makeParams(ngInputsPrc, {inputShapes.first});
std::shared_ptr<ngraph::Node> logicalNode;