|
tvm
|
Helpers for attribute objects. More...
#include <tvm/ffi/container/map.h>#include <tvm/ffi/extra/structural_equal.h>#include <tvm/ffi/extra/structural_hash.h>#include <tvm/ffi/function.h>#include <tvm/ffi/reflection/registry.h>#include <tvm/ir/cow.h>#include <string>#include <type_traits>#include <unordered_map>#include <utility>Go to the source code of this file.
Classes | |
| class | tvm::AttrsNode |
| Base class of all attribute class. More... | |
| class | tvm::Attrs |
| Managed reference to AttrsNode. 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... | |
Namespaces | |
| tvm | |
| An object that builds and maintains block scope and StmtSref mapping for Dependence analysis. | |
Functions | |
| 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... | |
Helpers for attribute objects.
This module enables declaration of named attributes which support default value setup and bound checking.