inference-engine/tests/helpers/ir_gen_helper.hpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #ifndef IR_GEN_HELPER_HPP
   6 #define IR_GEN_HELPER_HPP
   7
   8 #include "single_layer_common.hpp"
   9
  10 namespace single_layer_tests {
  11
  12     class IRTemplateGenerator {
  13         IRTemplateGenerator() = default;
  14     public:
  15         static std::string model_t;
  16
  17         static std::string getIRTemplate(const std::string& name,
  18                                   const std::vector<size_t>& input_shape,
  19                                   const std::string& precision,
  20                                   const std::string& layers,
  21                                   const std::string& edges,
  22                                   const unsigned ir_version = 5u);
  23     };
  24
  25 } // namespace single_layer_tests
  26 #endif /* IR_GEN_HELPER_HPP */
  27