summaryrefslogtreecommitdiff
path: root/inference-engine/tests/helpers/ir_gen_helper.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'inference-engine/tests/helpers/ir_gen_helper.hpp')
-rw-r--r--inference-engine/tests/helpers/ir_gen_helper.hpp27
1 files changed, 27 insertions, 0 deletions
diff --git a/inference-engine/tests/helpers/ir_gen_helper.hpp b/inference-engine/tests/helpers/ir_gen_helper.hpp
new file mode 100644
index 000000000..db8bff547
--- /dev/null
+++ b/inference-engine/tests/helpers/ir_gen_helper.hpp
@@ -0,0 +1,27 @@
+// Copyright (C) 2018 Intel Corporation
+// SPDX-License-Identifier: Apache-2.0
+//
+
+#ifndef IR_GEN_HELPER_HPP
+#define IR_GEN_HELPER_HPP
+
+#include "single_layer_common.hpp"
+
+namespace single_layer_tests {
+
+ class IRTemplateGenerator {
+ IRTemplateGenerator() = default;
+ public:
+ static std::string model_t;
+
+ static std::string getIRTemplate(const std::string& name,
+ const std::vector<size_t>& input_shape,
+ const std::string& precision,
+ const std::string& layers,
+ const std::string& edges,
+ const unsigned ir_version = 4u);
+ };
+
+} // namespace single_layer_tests
+#endif /* IR_GEN_HELPER_HPP */
+