[IE TESTS] Add VariadicSplit in SingleLayerTest (#1468)

* [IE TESTS] add variadic split

* [IE TESTS] update  variadic split

* [IE TESTS] update instance variadic split

* [IE TESTS] update variadic_split.cpp
This commit is contained in:
Anton Zaytsev
2020-08-05 15:06:55 +03:00
committed by GitHub
parent 0100a16228
commit 8570015347
5 changed files with 163 additions and 0 deletions

View File

@@ -0,0 +1,34 @@
// Copyright (C) 2019 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#pragma once
#include <tuple>
#include <string>
#include <vector>
#include <memory>
#include "functional_test_utils/layer_test_utils.hpp"
#include "ngraph_functions/builders.hpp"
namespace LayerTestsDefinitions {
typedef std::tuple<
std::vector<size_t>, // Num splits
size_t, // Axis
InferenceEngine::Precision, // Net precision
std::vector<size_t>, // Input shapes
std::string // Target device name
> VariadicSplitParams;
class VariadicSplitLayerTest : public testing::WithParamInterface<VariadicSplitParams>,
public LayerTestsUtils::LayerTestsCommon {
public:
static std::string getTestCaseName(testing::TestParamInfo<VariadicSplitParams> obj);
protected:
void SetUp() override;
};
} // namespace LayerTestsDefinitions