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, Map< String, Any > new_attrs) |
Copy the DictAttrs, but overrides attributes with the entries from attrs . More... | |
DictAttrs | tvm::WithAttr (DictAttrs attrs, 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, Map< 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.