|
tvm
|
ffi::Map<Op,ValueType> used to store meta-information about Op. More...
#include <op.h>
Public Types | |
| using | TParent = AttrRegistryMap< Op, ValueType > |
Public Member Functions | |
| ValueType | get (const RelaxExpr &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... | |
ffi::Map<Op,ValueType> used to store meta-information about Op.
| ValueType | The type of the value stored in map. |
| using tvm::OpAttrMap< ValueType >::TParent = AttrRegistryMap<Op, ValueType> |
|
inline |
get the corresponding value element at op with default value.
| expr | The key to the map |
| def_value | The default value when the key does not exist or if expr is not an Op. |
|
friend |