Map<Op,ValueType> used to store meta-information about Op. More...
#include <op.h>
Inheritance diagram for tvm::OpAttrMap< ValueType >:
Collaboration diagram for tvm::OpAttrMap< ValueType >:
Public Types | |
| using | TParent = AttrRegistryMap< Op, ValueType > |
Public Member Functions | |
| ValueType | get (const RelayExpr &expr, ValueType def_value) const |
| get the corresponding value element at op with default value. More... | |
 Public Member Functions inherited from tvm::AttrRegistryMap< Op, ValueType > | |
| AttrRegistryMap (const AttrRegistryMapContainerMap< Op > &map) | |
| constructor More... | |
| int | count (const Op &key) const |
| Check if the map has op as key. More... | |
| ValueType | operator[] (const Op &key) const |
| get the corresponding value element at key. More... | |
| ValueType | get (const Op &key, ValueType def_value) const |
| get the corresponding value element at key with default value. More... | |
Friends | |
| class | Op |
Additional Inherited Members | |
| Protected Attributes inherited from tvm::AttrRegistryMap< Op, ValueType > | |
| const AttrRegistryMapContainerMap< Op > & | map_ |
| The internal map field. More... | |
Detailed Description
template<typename ValueType>
class tvm::OpAttrMap< ValueType >
Map<Op,ValueType> used to store meta-information about Op.
- Template Parameters
-
ValueType The type of the value stored in map.
Member Typedef Documentation
◆ TParent
template<typename ValueType >
| using tvm::OpAttrMap< ValueType >::TParent = AttrRegistryMap<Op, ValueType> |
Member Function Documentation
◆ get()
template<typename ValueType >
|
inline |
get the corresponding value element at op with default value.
- Parameters
-
expr The key to the map def_value The default value when the key does not exist or if expr is not an Op.
- Returns
- the const reference to the content value.
Friends And Related Function Documentation
◆ Op
template<typename ValueType >
|
friend |
The documentation for this class was generated from the following file:
- include/tvm/ir/op.h
