Revise Split Single Layer Tests (#5190)

* Add serialization single layer tests

* Add more precisions, axis and num of splits in single layer tests

* Add op to script of trusted operations
This commit is contained in:
Gabriele Galiero Casay 2021-04-12 17:47:58 +02:00 committed by GitHub
parent 4337a2f235
commit c5d173c97a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 44 additions and 2 deletions

View File

@ -0,0 +1,39 @@
// Copyright (C) 2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include <vector>
#include "shared_test_classes/single_layer/split.hpp"
using namespace LayerTestsDefinitions;
namespace {
TEST_P(SplitLayerTest, Serialize) {
Serialize();
}
const std::vector<InferenceEngine::Precision> netPrecisions = {
InferenceEngine::Precision::FP32,
InferenceEngine::Precision::I32,
InferenceEngine::Precision::U16,
InferenceEngine::Precision::BOOL};
INSTANTIATE_TEST_CASE_P(
smoke_Split_Serialization, SplitLayerTest,
::testing::Combine(
::testing::Values(1, 2, 5, 10),
::testing::Values(0, 1, 2, 3),
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(std::vector<size_t>{20, 30, 50, 50}),
::testing::Values(std::vector<size_t>({})),
::testing::Values(CommonTestUtils::DEVICE_CPU)),
SplitLayerTest::getTestCaseName
);
} // namespace

View File

@ -13,12 +13,14 @@ namespace {
const std::vector<InferenceEngine::Precision> netPrecisions = {
InferenceEngine::Precision::FP32,
InferenceEngine::Precision::FP16
InferenceEngine::Precision::FP16,
InferenceEngine::Precision::I32,
InferenceEngine::Precision::U8
};
INSTANTIATE_TEST_CASE_P(smoke_NumSplitsCheck, SplitLayerTest,
::testing::Combine(
::testing::Values(1),
::testing::Values(1, 2, 3, 5),
::testing::Values(0, 1, 2, 3),
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),

View File

@ -91,6 +91,7 @@ verified_operations = [
'Sin-0',
'SoftPlus-4',
'Softmax-1',
'Split-1',
'StridedSlice-1',
'Substract-1',
'Swish-4',