1 // Copyright (C) 2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
12 #include <ie_layers.h>
14 namespace InferenceEngine {
15 namespace ShapeInfer {
18 *@brief Implementation of Const inference for TBD layer
20 class ConstConstInfer : public ConstInferImpl {
22 explicit ConstConstInfer(const std::string& type) : ConstInferImpl(type) {}
24 void inferImpl(const std::vector<Blob::CPtr>& inData,
25 const std::map<std::string, std::string>& params,
26 const std::map<std::string, Blob::Ptr>& blobs,
27 std::vector<Blob::Ptr>& outData) override {
28 auto it = blobs.find("custom");
29 if (it == blobs.end()) THROW_IE_EXCEPTION << "Missed `custom` blob";
30 // TODO: copy instead of putting pointer?
31 outData[0] = (*it).second;
35 } // namespace ShapeInfer
36 } // namespace InferenceEngine