tvm
Classes | Namespaces | Functions
attrs.h File Reference

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>
Include dependency graph for attrs.h:
This graph shows which files directly or indirectly include this file:

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...
 

Detailed Description

Helpers for attribute objects.

This module enables declaration of named attributes which support default value setup and bound checking.

See also
AttrsNode, TVM_DECLARE_ATTRS, TVM_ATTR_FIELD