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: 5328c8020a
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '5328c8020a'
${ noResults }
graphengine/inc/graph/utils/attr_utils.h

151 lines
9.0 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_GRAPH_UTILS_ATTR_UTILS_H_

  18. #define INC_GRAPH_UTILS_ATTR_UTILS_H_

  19. 

  20. #include <memory>

  21. #include <string>

  22. #include <vector>

  23. #include "graph/detail/attributes_holder.h"

  24. #include "graph/ge_attr_value.h"

  25. #include "graph/types.h"

  26. 

  27. namespace ge {

  28. class OpDesc;

  29. using OpDescPtr = std::shared_ptr<OpDesc>;

  30. using ConstOpDescPtr = std::shared_ptr<const OpDesc>;

  31. 

  32. class GE_FUNC_DEV_VISIBILITY GE_FUNC_HOST_VISIBILITY AttrUtils {

  33.  public:

  34.   class ConstAttrHolderAdapter;

  35.   class AttrHolderAdapter;

  36.   // Set

  37.   static bool HasAttr(ConstAttrHolderAdapter &&obj, const string &name);

  38. 

  39.   static bool SetInt(AttrHolderAdapter &&obj, const string &name, const int64_t &value);

  40.   static bool SetListInt(AttrHolderAdapter &&obj, const string &name, const vector<int64_t> &value);

  41.   static bool SetListInt(AttrHolderAdapter &&obj, const string &name, const vector<uint32_t> &value);

  42.   static bool SetListInt(AttrHolderAdapter &&obj, const string &name, const vector<int32_t> &value);

  43.   static bool SetListInt(AttrHolderAdapter &&obj, const string &name, std::initializer_list<int64_t> &&value);

  44. 

  45.   static bool SetFloat(AttrHolderAdapter &&obj, const string &name, const float &value);

  46.   static bool SetListFloat(AttrHolderAdapter &&obj, const string &name, const vector<float> &value);

  47.   static bool SetBool(AttrHolderAdapter &&obj, const string &name, const bool &value);

  48.   static bool SetListBool(AttrHolderAdapter &&obj, const string &name, const vector<bool> &value);

  49.   static bool SetStr(AttrHolderAdapter &&obj, const string &name, const string &value);

  50.   static bool SetListStr(AttrHolderAdapter &&obj, const string &name, const vector<string> &value);

  51.   static bool SetTensorDesc(AttrHolderAdapter &&obj, const string &name, const GeTensorDesc &value);

  52.   static bool SetListTensorDesc(AttrHolderAdapter &&obj, const string &name, const vector<GeTensorDesc> &value);

  53.   static bool SetTensor(AttrHolderAdapter &&obj, const string &name, const GeTensorPtr &value);

  54.   static bool SetTensor(AttrHolderAdapter &&obj, const string &name, const ConstGeTensorPtr &value);

  55.   static bool SetTensor(AttrHolderAdapter &&obj, const string &name, const GeTensor &value);

  56.   static bool SetListTensor(AttrHolderAdapter &&obj, const string &name, const vector<GeTensorPtr> &value);

  57.   static bool SetListTensor(AttrHolderAdapter &&obj, const string &name, const vector<ConstGeTensorPtr> &value);

  58.   static bool SetListTensor(AttrHolderAdapter &&obj, const string &name,

  59.                             std::initializer_list<ConstGeTensorPtr> &&value);

  60.   static bool SetListTensor(AttrHolderAdapter &&obj, const string &name, const vector<GeTensor> &value);

  61.   static bool SetGraph(AttrHolderAdapter &&obj, const string &name, const ComputeGraphPtr &value);

  62.   static bool SetListGraph(AttrHolderAdapter &&obj, const string &name, const vector<ComputeGraphPtr> &value);

  63.   static bool SetBytes(AttrHolderAdapter &&obj, const string &name, const GeAttrValue::BYTES &value);

  64.   static bool SetListBytes(AttrHolderAdapter &&obj, const string &name, const vector<GeAttrValue::BYTES> &value);

  65.   static bool SetNamedAttrs(AttrHolderAdapter &&obj, const string &name, const GeAttrValue::NAMED_ATTRS &value);

  66.   static bool SetListNamedAttrs(AttrHolderAdapter &&obj, const string &name,

  67.                                 const vector<GeAttrValue::NAMED_ATTRS> &value);

  68.   static bool SetListOpDesc(AttrHolderAdapter &&obj, const string &name, const vector<ConstOpDescPtr> &value);

  69.   static bool SetListOpDesc(AttrHolderAdapter &&obj, const string &name, const vector<OpDescPtr> &value);

  70. 

  71.   // Get

  72.   static bool GetInt(ConstAttrHolderAdapter &&obj, const string &name, int64_t &value);

  73.   static bool GetInt(ConstAttrHolderAdapter &&obj, const string &name, int32_t &value);

  74.   static bool GetInt(ConstAttrHolderAdapter &&obj, const string &name, uint32_t &value);

  75.   static bool GetListInt(ConstAttrHolderAdapter &&obj, const string &name, vector<int64_t> &value);

  76.   static bool GetListInt(ConstAttrHolderAdapter &&obj, const string &name, vector<int32_t> &value);

  77.   static bool GetListInt(ConstAttrHolderAdapter &&obj, const string &name, vector<uint32_t> &value);

  78.   static bool GetFloat(ConstAttrHolderAdapter &&obj, const string &name, float &value);

  79.   static bool GetListFloat(ConstAttrHolderAdapter &&obj, const string &name, vector<float> &value);

  80.   static bool GetBool(ConstAttrHolderAdapter &&obj, const string &name, bool &value);

  81.   static bool GetListBool(ConstAttrHolderAdapter &&obj, const string &name, vector<bool> &value);

  82.   static bool GetStr(ConstAttrHolderAdapter &&obj, const string &name, string &value);

  83.   static bool GetListStr(ConstAttrHolderAdapter &&obj, const string &name, vector<string> &value);

  84.   static bool GetTensorDesc(ConstAttrHolderAdapter &&obj, const string &name, GeTensorDesc &value);

  85.   static bool GetListTensorDesc(ConstAttrHolderAdapter &&obj, const string &name, vector<GeTensorDesc> &value);

  86.   static bool GetTensor(ConstAttrHolderAdapter &&obj, const string &name, ConstGeTensorPtr &value);

  87.   static bool MutableTensor(AttrHolderAdapter &&obj, const string &name, GeTensorPtr &value);

  88.   static bool GetListTensor(ConstAttrHolderAdapter &&obj, const string &name, vector<ConstGeTensorPtr> &value);

  89.   static bool MutableListTensor(AttrHolderAdapter &&obj, const string &name, vector<GeTensorPtr> &value);

  90.   static bool GetGraph(ConstAttrHolderAdapter &&obj, const string &name, ComputeGraphPtr &value);

  91.   static bool GetListGraph(ConstAttrHolderAdapter &&obj, const string &name, vector<ComputeGraphPtr> &value);

  92.   static bool GetBytes(ConstAttrHolderAdapter &&obj, const string &name, GeAttrValue::BYTES &value);

  93.   static bool GetListBytes(ConstAttrHolderAdapter &&obj, const string &name, vector<GeAttrValue::BYTES> &value);

  94.   static bool GetNamedAttrs(ConstAttrHolderAdapter &&obj, const string &name, GeAttrValue::NAMED_ATTRS &value);

  95.   static bool GetListNamedAttrs(ConstAttrHolderAdapter &&obj, const string &name,

  96.                                 vector<GeAttrValue::NAMED_ATTRS> &value);

  97.   static bool GetListOpDesc(ConstAttrHolderAdapter &&obj, const string &name, vector<OpDescPtr> &value);

  98.   // Value will be moved

  99.   static bool SetZeroCopyBytes(AttrHolderAdapter &&obj, const string &name, Buffer &&buffer);

  100.   static bool GetZeroCopyBytes(ConstAttrHolderAdapter &&obj, const string &name, Buffer &buffer);

  101.   // Value will be moved

  102.   static bool SetZeroCopyListBytes(AttrHolderAdapter &&obj, const string &name, vector<Buffer> &listBuffer);

  103.   static bool GetZeroCopyListBytes(ConstAttrHolderAdapter &&obj, const string &name, vector<Buffer> &listBuffer);

  104. 

  105.   static bool SetListListInt(AttrHolderAdapter &&obj, const string &name, const vector<vector<int64_t>> &value);

  106.   static bool GetListListInt(ConstAttrHolderAdapter &&obj, const string &name, vector<vector<int64_t>> &value);

  107. 

  108.   static bool SetListDataType(AttrHolderAdapter &&obj, const string &name, const vector<ge::DataType> &value);

  109.   static bool GetListDataType(ConstAttrHolderAdapter &&obj, const string &name, vector<ge::DataType> &value);

  110. 

  111.   static bool SetDataType(AttrHolderAdapter &&obj, const string &name, const ge::DataType &value);

  112.   static bool GetDataType(ConstAttrHolderAdapter &&obj, const string &name, ge::DataType &value);

  113. 

  114.   static OpDescPtr CloneOpDesc(const ConstOpDescPtr &orgOpDesc);

  115. 

  116.   static OpDescPtr CopyOpDesc(const ConstOpDescPtr &orgOpDesc);

  117. 

  118.   static std::string GetAllAttrsStr(ConstAttrHolderAdapter &&obj);

  119. 

  120.   class AttrHolderAdapter {

  121.    public:

  122.     AttrHolderAdapter(AttrHolder *obj) : obj_(obj) {}

  123.     ~AttrHolderAdapter() {}

  124.     template <class T>

  125.     AttrHolderAdapter(const std::shared_ptr<T> &obj) : obj_(obj.get()) {}

  126.     AttrHolderAdapter(AttrHolder &obj) : obj_(&obj) {}

  127.     operator bool() const { return obj_ != nullptr; }

  128.     AttrHolder *operator->() { return obj_; }

  129.     AttrHolder *get() { return obj_; }

  130. 

  131.     AttrHolder *obj_;

  132.   };

  133. 

  134.   class ConstAttrHolderAdapter {

  135.    public:

  136.     ConstAttrHolderAdapter(const AttrHolder *obj) : obj_(obj) {}

  137.     ~ConstAttrHolderAdapter() {}

  138.     template <class T>

  139.     ConstAttrHolderAdapter(const std::shared_ptr<T> obj) : obj_(obj.get()) {}

  140.     ConstAttrHolderAdapter(const AttrHolder &obj) : obj_(&obj) {}

  141.     operator bool() const { return obj_ != nullptr; }

  142.     const AttrHolder *operator->() const { return obj_; }

  143.     const AttrHolder *get() const { return obj_; }

  144. 

  145.    private:

  146.     const AttrHolder *obj_;

  147.   };

  148. };

  149. }  // namespace ge

  150. #endif  // INC_GRAPH_UTILS_ATTR_UTILS_H_