This class is the main extension interface. More...
#include <ie_iextension.h>
Public Member Functions | |
virtual StatusCode | getFactoryFor (ILayerImplFactory *&factory, const CNNLayer *cnnLayer, ResponseDesc *resp) noexcept |
Provides a factory for a specified CNNLayer. More... | |
virtual StatusCode | getPrimitiveTypes (char **&types, unsigned int &size, ResponseDesc *resp) noexcept |
Fills passed array with types of layers which kernel implementations are included in the extension. More... | |
StatusCode | getShapeInferTypes (char **&, unsigned int &, ResponseDesc *) noexcept override |
Fills passed array with types of layers which shape infer implementations are included in the extension. More... | |
StatusCode | getShapeInferImpl (IShapeInferImpl::Ptr &, const char *, ResponseDesc *) noexcept override |
Gets shape propagation implementation for the given string-type of CNNLayer. More... | |
virtual std::map< std::string, ngraph::OpSet > | getOpSets () |
Returns operation sets This method throws an exception if it was not implemented. More... | |
virtual std::vector< std::string > | getImplTypes (const std::shared_ptr< ngraph::Node > &node) |
Returns vector of implementation types. More... | |
virtual ILayerImpl::Ptr | getImplementation (const std::shared_ptr< ngraph::Node > &node, const std::string &implType) |
Returns implementation for specific nGraph op. More... | |
Public Member Functions inherited from InferenceEngine::IShapeInferExtension | |
virtual void | SetLogCallback (InferenceEngine::IErrorListener &listener) noexcept |
Sets logging callback. More... | |
virtual void | GetVersion (const InferenceEngine::Version *&versionInfo) const noexcept=0 |
Gets extension version information and stores in versionInfo. More... | |
virtual void | Unload () noexcept=0 |
Cleans resources up. | |
This class is the main extension interface.
|
inlinevirtualnoexcept |
Provides a factory for a specified CNNLayer.
factory | A factory returned from an extension plugin |
cnnLayer | A CNNLayer object to provide factory for |
resp | Response descriptor |
Reimplemented in InferenceEngine::Extension.
|
inlinevirtual |
Returns implementation for specific nGraph op.
node | shared pointer to nGraph op |
implType | implementation type |
Reimplemented in InferenceEngine::Extension, and TemplateExtension::Extension.
|
inlinevirtual |
Returns vector of implementation types.
node | shared pointer to nGraph op |
Reimplemented in InferenceEngine::Extension, and TemplateExtension::Extension.
|
virtual |
Returns operation sets This method throws an exception if it was not implemented.
Reimplemented in InferenceEngine::Extension, and TemplateExtension::Extension.
|
inlinevirtualnoexcept |
Fills passed array with types of layers which kernel implementations are included in the extension.
types | Array to store the layer types |
size | Size of the layer types array |
resp | Response descriptor |
Reimplemented in InferenceEngine::Extension.
|
inlineoverridevirtualnoexcept |
Gets shape propagation implementation for the given string-type of CNNLayer.
impl | the vector with implementations which is ordered by priority |
type | A type of CNNLayer |
resp | response descriptor |
Implements InferenceEngine::IShapeInferExtension.
|
inlineoverridevirtualnoexcept |
Fills passed array with types of layers which shape infer implementations are included in the extension.
types | Array to store the layer types |
size | Size of the layer types array |
resp | Response descriptor |
Implements InferenceEngine::IShapeInferExtension.