inference-engine/src/inference_engine/shape_infer/built-in/ie_built_in_holder.hpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #pragma once
   6
   7 #include <string>
   8 #include <vector>
   9 #include <list>
  10 #include <map>
  11 #include <memory>
  12
  13 #include <ie_iextension.h>
  14 #include "details/caseless.hpp"
  15 #include <description_buffer.hpp>
  16
  17 namespace InferenceEngine {
  18 namespace ShapeInfer {
  19
  20 /**
  21  *@brief Holder of shape infer implementations for build-in IE layers, that plugins support out-of-the-box
  22  */
  23 class INFERENCE_ENGINE_API_CLASS(BuiltInShapeInferHolder) : public IShapeInferExtension {
  24     struct ImplsHolder {
  25         using Ptr = std::shared_ptr<ImplsHolder>;
  26         InferenceEngine::details::caseless_map<std::string, IShapeInferImpl::Ptr> list;
  27     };
  28 public:
  29     StatusCode getShapeInferTypes(char**& types, unsigned int& size, ResponseDesc* resp) noexcept override;
  30
  31     StatusCode getShapeInferImpl(IShapeInferImpl::Ptr& impl, const char* type, ResponseDesc* resp) noexcept override;
  32
  33     void GetVersion(const InferenceEngine::Version*& versionInfo) const noexcept override {}
  34
  35     void Release() noexcept override { delete this; };
  36
  37     void Unload() noexcept override {};
  38
  39     static void AddImpl(const std::string& name, const IShapeInferImpl::Ptr& impl);
  40
  41     void SetLogCallback(InferenceEngine::IErrorListener& listener) noexcept override;
  42
  43 private:
  44     static ImplsHolder::Ptr GetImplsHolder();
  45 };
  46
  47 }  // namespace ShapeInfer
  48 }  // namespace InferenceEngine