tvm
Loading...
Searching...
No Matches
Classes | Namespaces | Functions | Variables
expr.h File Reference

Base expr nodes in TVM. More...

#include <tvm/ffi/dtype.h>
#include <tvm/ffi/extra/dataclass.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ffi/string.h>
#include <tvm/ir/attrs.h>
#include <tvm/ir/base_expr.h>
#include <tvm/ir/cow.h>
#include <tvm/ir/source_map.h>
#include <algorithm>
#include <functional>
#include <limits>
#include <optional>
#include <string>
Include dependency graph for expr.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::TupleNode
 Tuple container. More...
 
class  tvm::Tuple
 Managed reference to TupleNode. More...
 
class  tvm::TupleGetItemNode
 Get the index-th field out of a tuple. More...
 
class  tvm::TupleGetItem
 Managed reference to TupleGetItemNode. More...
 
class  tvm::TensorLoadNode
 Load a value from an indexed expression source. More...
 
class  tvm::TensorLoad
 Managed reference to TensorLoadNode. More...
 
class  tvm::VarNode
 A local variable in the IR. More...
 
class  tvm::Var
 Managed reference to VarNode. More...
 
class  tvm::GlobalVarNode
 Global variable that lives in the top-level module. More...
 
class  tvm::GlobalVar
 Managed reference to GlobalVarNode. More...
 
class  tvm::CallNode
 Call corresponds to callable invocation. More...
 
class  tvm::Call
 Managed reference to CallNode. More...
 
class  tvm::IntImmNode
 Constant integer literals in the program. More...
 
class  tvm::IntImm
 Managed reference class to IntImmNode. More...
 
class  tvm::FloatImmNode
 Constant floating point literals in the program. More...
 
class  tvm::FloatImm
 Managed reference class to FloatImmNode. More...
 
class  tvm::RangeNode
 range over one dimension More...
 
class  tvm::Range
 Range container
More...
 
struct  tvm::ffi::TypeTraits< IntImm >
 
struct  tvm::ffi::TypeTraits< FloatImm >
 

Namespaces

namespace  tvm
 An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
 
namespace  tvm::ffi
 

Functions

PrimExpr tvm::operator+ (PrimExpr a, PrimExpr b)
 add operator
 
PrimExpr tvm::operator- (PrimExpr a, PrimExpr b)
 subtraction operator
 
PrimExpr tvm::operator- (PrimExpr a)
 negation.
 
PrimExpr tvm::operator* (PrimExpr a, PrimExpr b)
 multiplication operator
 
PrimExpr tvm::operator/ (PrimExpr a, PrimExpr b)
 division operator
 
PrimExpr tvm::operator<< (PrimExpr a, PrimExpr b)
 left shift operator
 
PrimExpr tvm::operator>> (PrimExpr a, PrimExpr b)
 right shift operator
 
PrimExpr tvm::operator> (PrimExpr a, PrimExpr b)
 greater
 
PrimExpr tvm::operator>= (PrimExpr a, PrimExpr b)
 greater_equal
 
PrimExpr tvm::operator< (PrimExpr a, PrimExpr b)
 less
 
PrimExpr tvm::operator<= (PrimExpr a, PrimExpr b)
 less_equal
 
PrimExpr tvm::operator== (PrimExpr a, PrimExpr b)
 equal
 
PrimExpr tvm::operator!= (PrimExpr a, PrimExpr b)
 not_equal
 
PrimExpr tvm::operator&& (PrimExpr a, PrimExpr b)
 and
 
PrimExpr tvm::operator|| (PrimExpr a, PrimExpr b)
 or
 
PrimExpr tvm::operator! (PrimExpr a)
 not
 
PrimExpr tvm::operator& (PrimExpr a, PrimExpr b)
 take bitwise and of two values
 
PrimExpr tvm::operator| (PrimExpr a, PrimExpr b)
 take bitwise or of two values
 
PrimExpr tvm::operator^ (PrimExpr a, PrimExpr b)
 take bitwise xor of two values
 
PrimExpr tvm::operator~ (PrimExpr a)
 take bitwise negation of two values
 

Variables

template<>
constexpr bool tvm::ffi::object_ref_contains_v< PrimExpr, IntImmNode > = true
 
template<>
constexpr bool tvm::ffi::object_ref_contains_v< PrimExpr, FloatImmNode > = true
 
template<>
constexpr bool tvm::ffi::object_ref_contains_v< PrimExpr, TensorLoadNode > = true
 
template<>
constexpr bool tvm::ffi::use_default_type_traits_v< IntImm > = false
 
template<>
constexpr bool tvm::ffi::use_default_type_traits_v< FloatImm > = false
 

Detailed Description

Base expr nodes in TVM.