|
tvm
|
Helpers for attribute objects. More...
#include <dmlc/common.h>#include <tvm/ffi/container/map.h>#include <tvm/ffi/function.h>#include <tvm/ffi/reflection/accessor.h>#include <tvm/ffi/reflection/registry.h>#include <tvm/ir/expr.h>#include <tvm/node/structural_equal.h>#include <tvm/node/structural_hash.h>#include <functional>#include <string>#include <type_traits>#include <unordered_map>#include <utility>#include <vector>Go to the source code of this file.
Classes | |
| class | tvm::AttrFieldInfoNode |
| Information about attribute fields in string representations. More... | |
| class | tvm::AttrFieldInfo |
| AttrFieldInfo. More... | |
| class | tvm::BaseAttrsNode |
| Base class of all attribute class. More... | |
| class | tvm::Attrs |
| Managed reference to BaseAttrsNode. More... | |
| class | tvm::DictAttrsNode |
| Specialized attribute type that is backed by a map. The DictAttrsNode implements the Attrs behavior, its fields are directly accessible via object.field_name like other normal nodes. More... | |
| class | tvm::DictAttrs |
| Managed reference to DictAttrsNode. More... | |
| class | tvm::AttrsNodeReflAdapter< DerivedType > |
| Adapter for AttrsNode with the new reflection API. More... | |
Namespaces | |
| tvm | |
| Performance counters for profiling via the PAPI library. | |
Functions | |
| template<typename TObjectRef > | |
| TObjectRef | tvm::NullValue () |
| Create a NodeRef type that represents null. More... | |
| template<> | |
| DataType | tvm::NullValue< DataType > () |
| DictAttrs | tvm::WithAttrs (DictAttrs attrs, ffi::Map< ffi::String, Any > new_attrs) |
Copy the DictAttrs, but overrides attributes with the entries from attrs. More... | |
| DictAttrs | tvm::WithAttr (DictAttrs attrs, ffi::String key, Any value) |
| Copy the DictAttrs, but overrides a single attribute. More... | |
| DictAttrs | tvm::WithAttr (DictAttrs attrs, const std::string &key, Any value) |
| DictAttrs | tvm::WithoutAttr (DictAttrs attrs, const std::string &key) |
| Copy the DictAttrs, but without a specific attribute. More... | |
| template<typename TFunc > | |
| TFunc | tvm::WithAttr (TFunc input, const std::string &attr_key, Any attr_value) |
| Copy the function or module, but overrides the attribute value key with the value. More... | |
| template<typename TFunc > | |
| TFunc | tvm::WithAttrs (TFunc input, ffi::Map< ffi::String, Any > attrs) |
Copy the function or module, but overrides the attributes with the entries from attrs. More... | |
| template<typename TFunc > | |
| TFunc | tvm::WithoutAttr (TFunc input, const std::string &attr_key) |
| Copy the function or module, but removes the specified attribute. More... | |
| template<typename TAttrs > | |
| TAttrs | tvm::AttrsWithDefaultValues () |
| Create an Attr object with all default values. More... | |
Helpers for attribute objects.
This module enables declaration of named attributes which support default value setup and bound checking.