hummingbird
/
graphengine
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 121b9f7a21
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '121b9f7a21'
${ noResults }
graphengine/inc/framework/common/op/attr_value_util.h

161 lines
9.4 kB
Raw Blame History 

  1. /**

  2.  * Copyright 2019-2020 Huawei Technologies Co., Ltd

  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. #ifndef INC_FRAMEWORK_COMMON_OP_ATTR_VALUE_UTIL_H_

  18. #define INC_FRAMEWORK_COMMON_OP_ATTR_VALUE_UTIL_H_

  19. 

  20. #include <google/protobuf/map.h>

  21. #include <unordered_map>

  22. #include <string>

  23. 

  24. #include "graph/debug/ge_attr_define.h"

  25. #include "proto/om.pb.h"

  26. 

  27. using domi::AttrDef;

  28. using domi::AttrDef_ListValue;

  29. using domi::ModelDef;

  30. using domi::NamedAttrs;

  31. using domi::OpDef;

  32. 

  33. namespace ge {

  34. using AttrDefMap = ::google::protobuf::Map<::std::string, ::domi::AttrDef>;

  35. using AttrDefPair = ::google::protobuf::MapPair<std::string, domi::AttrDef>;

  36. 

  37. void AddOpAttr(const std::string &key, AttrDef &attr, OpDef *opdef);

  38. // DEFINE_ADD_ATTR_VALUE

  39. void AddOpAttr(const std::string &key, const std::string &value, AttrDefMap *attrs);

  40. void AddOpAttr(const std::string &key, const char *value, AttrDefMap *attrs);

  41. void AddOpAttr(const char *key, const char *value, AttrDefMap *attrs);

  42. void AddOpAttr(const std::string &key, const uint32_t value, AttrDefMap *attrs);

  43. void AddOpAttr(const std::string &key, const int32_t value, AttrDefMap *attrs);

  44. void AddOpAttr(const std::string &key, const int64_t value, AttrDefMap *attrs);

  45. void AddOpAttr(const std::string &key, const float value, AttrDefMap *attrs);

  46. void AddOpAttr(const std::string &key, const double value, AttrDefMap *attrs);

  47. void AddOpAttr(const std::string &key, const bool value, AttrDefMap *attrs);

  48. 

  49. void AddOpAttr(const std::string &key, const AttrDef_ListValue &value, AttrDefMap *attrs);

  50. 

  51. // DEFINE_ADD_ATTR_VALUE

  52. void AddOpAttr(const std::string &key, const std::string &value, OpDef *opdef);

  53. void AddOpAttr(const std::string &key, const char *value, OpDef *opdef);

  54. void AddOpAttr(const char *key, const char *value, OpDef *opdef);

  55. void AddOpAttr(const std::string &key, const uint32_t value, OpDef *opdef);

  56. void AddOpAttr(const std::string &key, const int32_t value, OpDef *opdef);

  57. void AddOpAttr(const std::string &key, const int64_t value, OpDef *opdef);

  58. void AddOpAttr(const std::string &key, const float value, OpDef *opdef);

  59. void AddOpAttr(const std::string &key, const double value, OpDef *opdef);

  60. void AddOpAttr(const std::string &key, const bool value, OpDef *opdef);

  61. 

  62. void AddOpAttr(const std::string &key, const AttrDef_ListValue &value, OpDef *opdef);

  63. 

  64. void AddOpBytesAttr(const std::string &key, const void *value, size_t size, OpDef *opdef);

  65. 

  66. // DEFINE_ADD_ATTR_VALUE_LIST

  67. void AddOpAttrList(const std::string &key, const double value, AttrDefMap *attrs);

  68. void AddOpAttrList(const std::string &key, const float value, AttrDefMap *attrs);

  69. void AddOpAttrList(const std::string &key, const uint32_t value, AttrDefMap *attrs);

  70. void AddOpAttrList(const std::string &key, const int32_t value, AttrDefMap *attrs);

  71. void AddOpAttrList(const std::string &key, const std::string value, AttrDefMap *attrs);

  72. void AddOpAttrList(const std::string &key, const double value, OpDef *opdef);

  73. void AddOpAttrList(const std::string &key, const float value, OpDef *opdef);

  74. void AddOpAttrList(const std::string &key, const uint32_t value, OpDef *opdef);

  75. void AddOpAttrList(const std::string &key, const int32_t value, OpDef *opdef);

  76. void AddOpAttrList(const std::string &key, const bool value, OpDef *opdef);

  77. void AddOpAttrList(const std::string &key, const int64_t value, OpDef *opdef);

  78. 

  79. void AddOpAttrList(const std::string &key, const std::string &value, OpDef *opdef);

  80. 

  81. bool GetOpAttr(const std::string &key, std::string *value, const OpDef *opdef);

  82. bool GetOpAttr(const std::string &key, int32_t *value, const OpDef *opdef);

  83. bool GetOpAttr(const std::string &key, int64_t *value, const OpDef *opdef);

  84. bool GetOpAttr(const std::string &key, uint32_t *value, const OpDef *opdef);

  85. bool GetOpAttr(const std::string &key, float *value, const OpDef *opdef);

  86. bool GetOpAttr(const std::string &key, double *value, const OpDef *opdef);

  87. bool GetOpAttr(const std::string &key, bool *value, const OpDef *opdef);

  88. bool GetOpAttr(const std::string &key, AttrDef_ListValue *value, const OpDef *opdef);

  89. 

  90. uint32_t GetOpAttrListSize(const std::string &key, std::string value, const OpDef *opdef);

  91. uint32_t GetOpAttrListSize(const std::string &key, int32_t value, const OpDef *opdef);

  92. uint32_t GetOpAttrListSize(const std::string &key, int64_t value, const OpDef *opdef);

  93. uint32_t GetOpAttrListSize(const std::string &key, uint32_t value, const OpDef *opdef);

  94. uint32_t GetOpAttrListSize(const std::string &key, float value, const OpDef *opdef);

  95. uint32_t GetOpAttrListSize(const std::string &key, double value, const OpDef *opdef);

  96. uint32_t GetOpAttrListSize(const std::string &key, bool value, const OpDef *opdef);

  97. 

  98. bool GetBytesAttr(const std::string &key, std::string *value, const OpDef *opdef);

  99. bool GetBytesAttr(const std::string &key, std::string *value, const ModelDef *model_def);

  100. 

  101. void AddModelAttr(const std::string &key, const std::string &value, ModelDef *model_def);

  102. void AddModelAttr(const std::string &key, const char *value, ModelDef *model_def);

  103. void AddModelAttr(const char *key, const char *value, ModelDef *model_def);

  104. void AddModelAttr(const std::string &key, const uint32_t value, ModelDef *model_def);

  105. void AddModelAttr(const std::string &key, const int32_t value, ModelDef *model_def);

  106. void AddModelAttr(const std::string &key, const int64_t value, ModelDef *model_def);

  107. void AddModelAttr(const std::string &key, const float value, ModelDef *model_def);

  108. void AddModelAttr(const std::string &key, const double value, ModelDef *model_def);

  109. void AddModelAttr(const std::string &key, const bool value, ModelDef *model_def);

  110. void AddModelAttr(const std::string &key, const void *value, size_t size, ModelDef *model_def);

  111. void AddModelAttr(const std::string &key, const AttrDef_ListValue &value, ModelDef *model_def);

  112. 

  113. void AddModelAttrList(const std::string &key, const double value, ModelDef *model_def);

  114. void AddModelAttrList(const std::string &key, const float value, ModelDef *model_def);

  115. void AddModelAttrList(const std::string &key, const uint32_t value, ModelDef *model_def);

  116. void AddModelAttrList(const std::string &key, const int32_t value, ModelDef *model_def);

  117. void AddModelAttrList(const std::string &key, const std::string &value, ModelDef *model_def);

  118. 

  119. bool GetModelAttr(const std::string &key, std::string *value, const ModelDef *model_def);

  120. bool GetModelAttr(const std::string &key, int32_t *value, const ModelDef *model_def);

  121. bool GetModelAttr(const std::string &key, int64_t *value, const ModelDef *model_def);

  122. bool GetModelAttr(const std::string &key, uint32_t *value, const ModelDef *model_def);

  123. bool GetModelAttr(const std::string &key, float *value, const ModelDef *model_def);

  124. bool GetModelAttr(const std::string &key, double *value, const ModelDef *model_def);

  125. bool GetModelAttr(const std::string &key, bool *value, const ModelDef *model_def);

  126. bool GetModelAttr(const std::string &key, AttrDef_ListValue *value, const ModelDef *model_def);

  127. 

  128. bool HasOpAttr(const OpDef *opdef, const std::string &attr_name);

  129. 

  130. void SetAttrDef(const std::string &value, AttrDef *out);

  131. void SetAttrDef(const char *value, AttrDef *out);

  132. void SetAttrDef(const uint32_t value, AttrDef *out);

  133. void SetAttrDef(const int32_t value, AttrDef *out);

  134. void SetAttrDef(const float value, AttrDef *out);

  135. void SetAttrDef(const double value, AttrDef *out);

  136. void SetAttrDef(const bool value, AttrDef *out);

  137. void SetAttrList(const std::string &value, AttrDef *out);

  138. void SetAttrList(const bool value, AttrDef *out);

  139. void SetAttrList(const float value, AttrDef *out);

  140. void SetAttrList(const double value, AttrDef *out);

  141. void SetAttrList(const uint32_t value, AttrDef *out);

  142. 

  143. bool GetAttrDefValue(const std::string &key, std::string *value, const AttrDefMap &attr);

  144. bool GetAttrDefValue(const std::string &key, int32_t *value, const AttrDefMap &attr);

  145. bool GetAttrDefValue(const std::string &key, int64_t *value, const AttrDefMap &attr);

  146. bool GetAttrDefValue(const std::string &key, uint32_t *value, const AttrDefMap &attr);

  147. bool GetAttrDefValue(const std::string &key, float *value, const AttrDefMap &attr);

  148. bool GetAttrDefValue(const std::string &key, double *value, const AttrDefMap &attr);

  149. bool GetAttrDefValue(const std::string &key, bool *value, const AttrDefMap &attr);

  150. bool GetAttrDefValue(const std::string &key, AttrDef_ListValue *value, const AttrDefMap &attr);

  151. bool GetAttrDefValue(const std::string &key, NamedAttrs *&value, AttrDefMap *attr);

  152. bool GetAttrDefValue(const std::string &key, const NamedAttrs *&value, const AttrDefMap &attr);

  153. 

  154. bool GetAttrDefListValue(const std::string &key, int idx, int32_t *value, const AttrDefMap &attr);

  155. bool GetAttrDefListValue(const std::string &key, int idx, uint32_t *value, const AttrDefMap &attr);

  156. bool GetAttrDefListValue(const std::string &key, int idx, float *value, const AttrDefMap &attr);

  157. bool GetAttrDefListValue(const std::string &key, int idx, double *value, const AttrDefMap &attr);

  158. }

  159. 

  160. #endif  // INC_FRAMEWORK_COMMON_OP_ATTR_VALUE_UTIL_H_