ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
COpCall.cpp
Go to the documentation of this file.
1/*
2 * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include "locoex/COpCall.h"
18
19#include "locoex/COpAttrTypes.h"
20
21namespace locoex
22{
23
24template <COpAttrType AT>
25const typename AttrTypeTrait<AT>::Type *COpCall::attr(const std::string &attr_name) const
26{
27 COpAttrData *attr_data;
28 auto found = _attrs.find(attr_name);
29 if (found != _attrs.end())
30 {
31 attr_data = found->second.get();
32 return dynamic_cast<const typename AttrTypeTrait<AT>::Type *>(attr_data);
33 }
34 else
35 throw std::runtime_error("Cannot find requested attr");
36}
37
38void COpCall::attr(const std::string &attr_name, std::unique_ptr<COpAttrData> &&attr_data)
39{
40 if (_attrs.find(attr_name) == _attrs.end())
41 _attrs[attr_name] = std::move(attr_data);
42 else
43 throw std::runtime_error("Attr already inserted");
44}
45
46std::vector<std::string> COpCall::attr_names() const
47{
48 std::vector<std::string> attr_names;
49
50 for (auto it = _attrs.cbegin(); it != _attrs.cend(); ++it)
51 {
52 attr_names.emplace_back(it->first);
53 }
54
55 return attr_names;
56}
57
58#define INSTANTIATE(AT) \
59 template const typename AttrTypeTrait<AT>::Type *COpCall::attr<AT>(const std::string &attr_name) \
60 const;
61
64
65#undef INSTANTIATE
66
67} // namespace locoex
#define INSTANTIATE(DT)
Definition Tensor.cpp:74
std::vector< std::string > attr_names() const
get all the names of attr
Definition COpCall.cpp:46
void attr(const std::string &attr_name, std::unique_ptr< COpAttrData > &&attr_data)
Store [attr_name, attr_data].
Definition COpCall.cpp:38
Struct that holds attr type.