diff options
Diffstat (limited to 'inference-engine/tests/helpers/ir_gen_helper.hpp')
-rw-r--r-- | inference-engine/tests/helpers/ir_gen_helper.hpp | 27 |
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 */ + |