A map data structure that takes Op* as key and returns ValueType.
More...
#include <op.h>
|
const ValueType & | operator[] (const Op *op) const |
| get the corresponding value element at op More...
|
|
const ValueType & | get (const Op *op, const ValueType &def_value) const |
| get the corresponding value element at op with default value. More...
|
|
int | count (const Op *op) const |
| Check if the map has op as key. More...
|
|
bool | contains (const Op *op) const |
| Check if the map has op as key. More...
|
|
|
class | Op |
| operator structure from NNVM More...
|
|
template<typename ValueType>
class nnvm::OpMap< ValueType >
A map data structure that takes Op* as key and returns ValueType.
- Template Parameters
-
ValueType | The type of the value stored in map. |
template<typename ValueType >
Check if the map has op as key.
- Parameters
-
- Returns
- true if op is contained in map, false otherwise.
template<typename ValueType >
Check if the map has op as key.
- Parameters
-
- Returns
- 1 if op is contained in map, 0 otherwise.
template<typename ValueType >
const ValueType & nnvm::OpMap< ValueType >::get |
( |
const Op * |
op, |
|
|
const ValueType & |
def_value |
|
) |
| const |
|
inline |
get the corresponding value element at op with default value.
- Parameters
-
op | The key to the map |
def_value | The default value when the key does not exist. |
- Returns
- the const reference to the content value.
template<typename ValueType >
const ValueType & nnvm::OpMap< ValueType >::operator[] |
( |
const Op * |
op | ) |
const |
|
inline |
get the corresponding value element at op
- Parameters
-
- Returns
- the const reference to the content value.
template<typename ValueType>
operator structure from NNVM
The documentation for this class was generated from the following file:
- /work/mxnet/3rdparty/tvm/nnvm/include/nnvm/op.h