diff options
Diffstat (limited to 'inference-engine/tests/helpers/ir_gen_helper.cpp')
-rw-r--r-- | inference-engine/tests/helpers/ir_gen_helper.cpp | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/inference-engine/tests/helpers/ir_gen_helper.cpp b/inference-engine/tests/helpers/ir_gen_helper.cpp new file mode 100644 index 000000000..40a05c451 --- /dev/null +++ b/inference-engine/tests/helpers/ir_gen_helper.cpp @@ -0,0 +1,48 @@ +// Copyright (C) 2018 Intel Corporation +// SPDX-License-Identifier: Apache-2.0 +// + +#include "ir_gen_helper.hpp" + +namespace single_layer_tests { + + std::string IRTemplateGenerator::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) { + std::string model = model_t; + REPLACE_WITH_STR(model, "_NAME_", name); + REPLACE_WITH_NUM(model, "_IRv_", ir_version); + REPLACE_WITH_STR(model, "_PR_", precision); + + std::string s_dims; + for (auto& dim : input_shape) { + s_dims += "\n <dim>"; + s_dims += std::to_string(dim) + "</dim>"; + } + REPLACE_WITH_STR(model, "__SRC_DIMS__", s_dims); + REPLACE_WITH_STR(model, "_LAYERS_", layers); + REPLACE_WITH_STR(model, "_EDGES_", edges); + + return model; + } + + std::string IRTemplateGenerator::model_t = R"V0G0N( + <net name="_NAME_" version="_IRv_" precision="_PR_" batch="1"> + <layers> + <layer name="in1" type="Input" precision="_PR_" id="0"> + <output> + <port id="0">__SRC_DIMS__ + </port> + </output> + </layer> + _LAYERS_ + </layers> + <edges> + _EDGES_ + </edges> + </net> + )V0G0N"; +}
\ No newline at end of file |