blob: 1193d1ffe83725f85b2076f960e13eef17b24e2c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
// Copyright (C) 2018 Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0
//
#pragma once
#include "ie_built_in_holder.hpp"
#include <string>
#include <memory>
namespace InferenceEngine {
namespace ShapeInfer {
template<typename Impl> class ImplRegisterBase {
public:
explicit ImplRegisterBase(const std::string& type) {
BuiltInShapeInferHolder::AddImpl(type, std::make_shared<Impl>(type));
}
};
#define REG_SHAPE_INFER_FOR_TYPE(__prim, __type) \
static ImplRegisterBase<__prim> __reg__##__type(#__type)
} // namespace ShapeInfer
} // namespace InferenceEngine
|
