tvm
|
a named variable in TIR More...
#include <var.h>
Public Types | |
using | ContainerType = VarNode |
type indicate the container type More... | |
Public Types inherited from tvm::runtime::ObjectRef | |
using | ContainerType = Object |
type indicate the container type. More... | |
Public Member Functions | |
Var (ObjectPtr< Object > n) | |
Var (String name_hint="v", DataType dtype=DataType::Int(32), Span span=Span()) | |
Constructor. More... | |
Var (String name_hint, Type type_annotation, Span span=Span()) | |
Constructor which provides a more detailed type annotation. More... | |
Var | copy_with_name (const String &name) const |
Make a new copy of var with same type, but a different nam. More... | |
Var | copy_with_suffix (const String &suffix) const |
Make a new copy of var with same type, append suffix. More... | |
Var | copy_with_dtype (DataType dtype) const |
Make a new copy of the variable with specified dtype. More... | |
const VarNode * | operator-> () const |
Get pointer to the internal value. More... | |
const VarNode * | get () const |
Get pointer to the internal value. More... | |
Public Member Functions inherited from tvm::PrimExpr | |
PrimExpr (int32_t value) | |
construct from integer. More... | |
PrimExpr (float value) | |
construct from float. More... | |
DataType | dtype () const |
TVM_DEFINE_OBJECT_REF_METHODS (PrimExpr, BaseExpr, PrimExprNode) | |
Public Member Functions inherited from tvm::BaseExpr | |
TVM_DEFINE_OBJECT_REF_METHODS (BaseExpr, ObjectRef, BaseExprNode) | |
Public Member Functions inherited from tvm::runtime::ObjectRef | |
ObjectRef ()=default | |
default constructor More... | |
ObjectRef (ObjectPtr< Object > data) | |
Constructor from existing object ptr. More... | |
bool | same_as (const ObjectRef &other) const |
Comparator. More... | |
bool | operator== (const ObjectRef &other) const |
Comparator. More... | |
bool | operator!= (const ObjectRef &other) const |
Comparator. More... | |
bool | operator< (const ObjectRef &other) const |
Comparator. More... | |
bool | defined () const |
const Object * | get () const |
const Object * | operator-> () const |
bool | unique () const |
int | use_count () const |
template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>> | |
const ObjectType * | as () const |
Try to downcast the internal Object to a raw pointer of a corresponding type. More... | |
template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>> | |
Optional< ObjectRefType > | as () const |
Try to downcast the ObjectRef to a Optional<T> of the requested type. More... | |
Additional Inherited Members | |
Static Public Attributes inherited from tvm::runtime::ObjectRef | |
static constexpr bool | _type_is_nullable = true |
Protected Member Functions inherited from tvm::runtime::ObjectRef | |
Object * | get_mutable () const |
Static Protected Member Functions inherited from tvm::runtime::ObjectRef | |
template<typename T > | |
static T | DowncastNoCheck (ObjectRef ref) |
Internal helper function downcast a ref without check. More... | |
static void | FFIClearAfterMove (ObjectRef *ref) |
Clear the object ref data field without DecRef after we successfully moved the field. More... | |
template<typename ObjectType > | |
static ObjectPtr< ObjectType > | GetDataPtr (const ObjectRef &ref) |
Internal helper function get data_ as ObjectPtr of ObjectType. More... | |
Protected Attributes inherited from tvm::runtime::ObjectRef | |
ObjectPtr< Object > | data_ |
Internal pointer that backs the reference. More... | |
a named variable in TIR
using tvm::tir::Var::ContainerType = VarNode |
type indicate the container type
|
explicit |
name_hint | variable name |
dtype | data type |
span | The location of this object in the source code. |
Constructor which provides a more detailed type annotation.
name_hint | variable name. |
type_annotation | The type annotation. |
span | The location of this object in the source code. |
Make a new copy of the variable with specified dtype.
dtype | The specified dtype |
Make a new copy of var with same type, but a different nam.
name | The new name to be used. |
Make a new copy of var with same type, append suffix.
suffix | The suffix to be appended. |
|
inline |
Get pointer to the internal value.
|
inline |
Get pointer to the internal value.