inference-engine/tests/functional/plugin/shared/include/subgraph_tests/negative_memory_layer_offset.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     size_t,                            //Input size
  20     size_t,                            //Hidden size
  21     std::map<std::string, std::string> //Configuration
  22 > NegativeMemoryLayerOffsetTuple;
  23
  24 class NegativeMemoryOffsetTest
  25     : public testing::WithParamInterface<NegativeMemoryLayerOffsetTuple>,
  26     public LayerTestsUtils::LayerTestsCommon {
  27 private:
  28     void switchToNgraphFriendlyModel();
  29     std::vector<float> memory_init;
  30 public:
  31     static std::string getTestCaseName(const testing::TestParamInfo<NegativeMemoryLayerOffsetTuple>& obj);
  32 protected:
  33     void SetUp() override;
  34     void Run() override;
  35 };
  36 } // namespace LayerTestsDefinitions