812cd1076b
* java bindings added * bulid saples alongside with IE build * fixes * fixes * fixed comments * fixes
86 lines
3.0 KiB
C++
86 lines
3.0 KiB
C++
#include <map>
|
|
#include "inference_engine.hpp"
|
|
|
|
//
|
|
// resize_algorithm
|
|
//
|
|
static const std::map<int, InferenceEngine::ResizeAlgorithm> resize_alg_map = {
|
|
{0, InferenceEngine::ResizeAlgorithm::NO_RESIZE},
|
|
{1, InferenceEngine::ResizeAlgorithm::RESIZE_AREA},
|
|
{2, InferenceEngine::ResizeAlgorithm::RESIZE_BILINEAR}
|
|
};
|
|
|
|
//
|
|
// layout
|
|
//
|
|
static const std::map<int, InferenceEngine::Layout> layout_map = {
|
|
{0, InferenceEngine::Layout::ANY},
|
|
{1, InferenceEngine::Layout::NCHW},
|
|
{2, InferenceEngine::Layout::NHWC},
|
|
{3, InferenceEngine::Layout::NCDHW},
|
|
{4, InferenceEngine::Layout::NDHWC},
|
|
{64, InferenceEngine::Layout::OIHW},
|
|
{95, InferenceEngine::Layout::SCALAR},
|
|
{96, InferenceEngine::Layout::C},
|
|
{128, InferenceEngine::Layout::CHW},
|
|
{192, InferenceEngine::Layout::HW},
|
|
{193, InferenceEngine::Layout::NC},
|
|
{194, InferenceEngine::Layout::CN},
|
|
{200, InferenceEngine::Layout::BLOCKED}
|
|
};
|
|
|
|
//
|
|
// precision
|
|
//
|
|
static const std::map<int, InferenceEngine::Precision> precision_map = {
|
|
{255, InferenceEngine::Precision::UNSPECIFIED},
|
|
{0, InferenceEngine::Precision::MIXED},
|
|
{10, InferenceEngine::Precision::FP32},
|
|
{11, InferenceEngine::Precision::FP16},
|
|
{20, InferenceEngine::Precision::Q78},
|
|
{30, InferenceEngine::Precision::I16},
|
|
{40, InferenceEngine::Precision::U8},
|
|
{50, InferenceEngine::Precision::I8},
|
|
{60, InferenceEngine::Precision::U16},
|
|
{70, InferenceEngine::Precision::I32},
|
|
{72, InferenceEngine::Precision::I64},
|
|
{71, InferenceEngine::Precision::BIN},
|
|
{80, InferenceEngine::Precision::CUSTOM}
|
|
};
|
|
|
|
//
|
|
// wait_mode
|
|
//
|
|
static const std::map<int, InferenceEngine::IInferRequest::WaitMode> wait_mode_map = {
|
|
{-1, InferenceEngine::IInferRequest::WaitMode::RESULT_READY},
|
|
{0, InferenceEngine::IInferRequest::WaitMode::STATUS_ONLY}
|
|
};
|
|
|
|
//
|
|
// status_code
|
|
//
|
|
static const std::map<InferenceEngine::StatusCode, int> status_code_map = {
|
|
{InferenceEngine::StatusCode::OK, 0},
|
|
{InferenceEngine::StatusCode::GENERAL_ERROR, -1},
|
|
{InferenceEngine::StatusCode::NOT_IMPLEMENTED, -2},
|
|
{InferenceEngine::StatusCode::NETWORK_NOT_LOADED, -3},
|
|
{InferenceEngine::StatusCode::PARAMETER_MISMATCH, -4},
|
|
{InferenceEngine::StatusCode::NOT_FOUND, -5},
|
|
{InferenceEngine::StatusCode::OUT_OF_BOUNDS, -6},
|
|
{InferenceEngine::StatusCode::UNEXPECTED, -7},
|
|
{InferenceEngine::StatusCode::REQUEST_BUSY, -8},
|
|
{InferenceEngine::StatusCode::RESULT_NOT_READY, -9},
|
|
{InferenceEngine::StatusCode::NOT_ALLOCATED, -10},
|
|
{InferenceEngine::StatusCode::INFER_NOT_STARTED, -11},
|
|
{InferenceEngine::StatusCode::NETWORK_NOT_READ, -12},
|
|
};
|
|
|
|
//
|
|
// layer_status
|
|
//
|
|
static const std::map<InferenceEngine::InferenceEngineProfileInfo::LayerStatus, int> layer_status_map = {
|
|
{InferenceEngine::InferenceEngineProfileInfo::LayerStatus::NOT_RUN, 0},
|
|
{InferenceEngine::InferenceEngineProfileInfo::LayerStatus::OPTIMIZED_OUT, 1},
|
|
{InferenceEngine::InferenceEngineProfileInfo::LayerStatus::EXECUTED, 2},
|
|
};
|