inference-engine/tests/functional/plugin/shared/include/subgraph_tests/split_trivial_permute_concat.hpp

   1 // Copyright (C) 2020 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 #pragma once
   4
   5 #include <tuple>
   6 #include <string>
   7 #include <vector>
   8 #include <memory>
   9
  10 #include "functional_test_utils/layer_test_utils.hpp"
  11 #include "ngraph_functions/utils/ngraph_helpers.hpp"
  12 #include "ngraph_functions/builders.hpp"
  13
  14 namespace LayerTestsDefinitions {
  15
  16 typedef std::tuple<
  17     InferenceEngine::Precision,        //Network precision
  18     std::string,                       //Device name
  19     std::vector<size_t>,               //Input sizes
  20     size_t,                            //Split axis
  21     size_t,                            //Concat axis
  22     std::map<std::string, std::string> //Configuration
  23 > SplitTrivialPermuteConcatTuple;
  24
  25 class SplitTrivialPermuteConcatTest
  26     : public testing::WithParamInterface<SplitTrivialPermuteConcatTuple>,
  27     public LayerTestsUtils::LayerTestsCommon {
  28 public:
  29     static std::string getTestCaseName(const testing::TestParamInfo<SplitTrivialPermuteConcatTuple>& obj);
  30 protected:
  31     void SetUp() override;
  32 };
  33 } // namespace LayerTestsDefinitions