[IE Myriad] Fix layer tests for logical_and (#2622)
This commit is contained in:
committed by
GitHub
parent
d617f1c230
commit
5ce622f4f4
@@ -22,15 +22,9 @@ typedef std::map<std::string, std::string> Config;
|
||||
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;
|
||||
SetupParams();
|
||||
|
||||
auto ngInputsPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(ngInputsPrecision);
|
||||
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;
|
||||
@@ -80,7 +74,7 @@ INSTANTIATE_TEST_CASE_P(smoke_EltwiseLogicalInt,
|
||||
::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),
|
||||
@@ -95,7 +89,7 @@ INSTANTIATE_TEST_CASE_P(smoke_EltwiseLogicalNotInt,
|
||||
::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),
|
||||
|
||||
@@ -34,10 +34,18 @@ class LogicalLayerTest : public testing::WithParamInterface<LogicalTestParams>,
|
||||
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
|
||||
|
||||
@@ -60,20 +60,18 @@ InferenceEngine::Blob::Ptr LogicalLayerTest::GenerateInput(const InferenceEngine
|
||||
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;
|
||||
|
||||
Reference in New Issue
Block a user