OpenVINO™ Toolkit
ie_context.hpp
Go to the documentation of this file.
1
// Copyright (C) 2018-2020 Intel Corporation
2
// SPDX-License-Identifier: Apache-2.0
3
//
4
5
/**
6
* @brief This is a header file for the IE Context class
7
*
8
* @file ie_context.hpp
9
*/
10
#pragma once
11
12
#include <
ie_iextension.h
>
13
14
#include <
details/caseless.hpp
>
15
#include <map>
16
#include <string>
17
#include <vector>
18
19
namespace
InferenceEngine
{
20
21
/**
22
* @deprecated Use ngraph API instead.
23
* @brief This class implements object
24
*/
25
class
INFERENCE_ENGINE_NN_BUILDER_API_CLASS(Context) {
26
public
:
27
Context
();
28
29
/**
30
* @brief Registers extension within the context
31
*
32
* @param ext Pointer to already loaded extension
33
*/
34
void
addExtension(
const
IShapeInferExtensionPtr
& ext);
35
36
/**
37
* @brief Registers Shape Infer implementation within the Context
38
*
39
* @param type Layer type
40
* @param impl Shape Infer implementation
41
*/
42
IE_SUPPRESS_DEPRECATED_START
43
void
addShapeInferImpl(
const
std::string&
type
,
const
IShapeInferImpl::Ptr
& impl);
44
45
/**
46
* @brief Returns the shape infer implementation by layer type
47
*
48
* @param type Layer type
49
* @return Shape Infer implementation
50
*/
51
IShapeInferImpl::Ptr
getShapeInferImpl(
const
std::string& type);
52
53
private
:
54
details::caseless_map<std::string, IShapeInferImpl::Ptr> shapeInferImpls;
55
IE_SUPPRESS_DEPRECATED_END
56
};
57
58
}
// namespace InferenceEngine
InferenceEngine
Inference Engine API.
Definition:
ie_argmax_layer.hpp:15
InferenceEngine::IShapeInferExtensionPtr
std::shared_ptr< IShapeInferExtension > IShapeInferExtensionPtr
A shared pointer to a IShapeInferExtension interface.
Definition:
ie_iextension.h:366
InferenceEngine::Context
This class implements object.
Definition:
ie_context.hpp:25
InferenceEngine::IShapeInferImpl::Ptr
std::shared_ptr< IShapeInferImpl > Ptr
A shared pointer to a IShapeInferImpl object.
Definition:
ie_iextension.h:192
ie_iextension.h
This is a header file for Inference Engine Extension Interface.
caseless.hpp
InferenceEngine::type
std::string type
Layer type.
Definition:
ie_layers.h:47