expr.h File Reference

TIR expressions. More...

#include <tvm/ir/expr.h>
#include <tvm/node/functor.h>
#include <tvm/node/node.h>
#include <tvm/runtime/c_runtime_api.h>
#include <tvm/runtime/container/array.h>
#include <tvm/runtime/container/map.h>
#include <tvm/runtime/container/string.h>
#include <tvm/runtime/data_type.h>
#include <tvm/tir/buffer.h>
#include <tvm/tir/var.h>
#include <algorithm>
#include <iostream>
#include <limits>
#include <string>
#include <unordered_map>
#include <utility>

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::tir::StringImmNode
	String constants, only used in asserts. More...
class	tvm::tir::StringImm
	Managed reference to StringImmNode. More...
class	tvm::tir::CastNode
	Cast value from one data type to another. More...
class	tvm::tir::Cast
	Managed reference to CastNode. More...
class	tvm::tir::BinaryOpNode< T >
	Base template to implement binary ops. More...
class	tvm::tir::AddNode
	a + b More...
class	tvm::tir::Add
	Managed reference to AddNode. More...
class	tvm::tir::SubNode
	a - b More...
class	tvm::tir::Sub
	Managed reference to SubNode. More...
class	tvm::tir::MulNode
	a * b More...
class	tvm::tir::Mul
	Managed reference to MulNode. More...
class	tvm::tir::DivNode
	a / b in the C semnatics. More...
class	tvm::tir::Div
	Managed reference to DivNode. More...
class	tvm::tir::ModNode
	a % b in the C semnatics. More...
class	tvm::tir::Mod
	Managed reference to ModNode. More...
class	tvm::tir::FloorDivNode
	Floor division, floor(a/b) More...
class	tvm::tir::FloorDiv
	Managed reference to FloorDivNode. More...
class	tvm::tir::FloorModNode
	The remainder of the floordiv. More...
class	tvm::tir::FloorMod
	Managed reference to FloorModNode. More...
class	tvm::tir::MinNode
	min(a, b) More...
class	tvm::tir::Min
	Managed reference to MinNode. More...
class	tvm::tir::MaxNode
	max(a, b) More...
class	tvm::tir::Max
	Managed reference to MaxNode. More...
class	tvm::tir::CmpOpNode< T >
	Base template to implement comparison ops. More...
class	tvm::tir::EQNode
	a == b More...
class	tvm::tir::EQ
	Managed reference to EQNode. More...
class	tvm::tir::NENode
	a != b More...
class	tvm::tir::NE
	Managed reference to NENode. More...
class	tvm::tir::LTNode
	a < b More...
class	tvm::tir::LT
	Managed reference to LTNode. More...
struct	tvm::tir::LENode
	a <= b More...
class	tvm::tir::LE
	Managed reference to LENode. More...
class	tvm::tir::GTNode
	a > b More...
class	tvm::tir::GT
	Managed reference to GTNode. More...
class	tvm::tir::GENode
	a >= b More...
class	tvm::tir::GE
	Managed reference to GENode. More...
class	tvm::tir::AndNode
	a && b More...
class	tvm::tir::And
	Managed reference to AndNode. More...
class	tvm::tir::OrNode
	a || b More...
class	tvm::tir::Or
	Managed reference to OrNode. More...
class	tvm::tir::NotNode
	!a More...
class	tvm::tir::Not
	Managed reference to NotNode. More...
class	tvm::tir::SelectNode
	return true_value if condition is true, otherwise return false_value. More...
class	tvm::tir::Select
	Managed reference to SelectNode. More...
class	tvm::tir::BufferLoadNode
	Load value from the high dimension buffer. More...
class	tvm::tir::BufferLoad
	Managed reference to BufferLoadNode. More...
class	tvm::tir::ProducerLoadNode
	Load value from the result produced by the producer. More...
class	tvm::tir::ProducerLoad
	Managed reference to ProducerLoadNode. More...
class	tvm::tir::RampNode
	Construct a vector with lanes elements where its i-th element equals base + i * stride. This is useful to construct a index for a continuous vector load. More...
class	tvm::tir::Ramp
	Managed reference to RampNode. More...
class	tvm::tir::BroadcastNode
	Create a vector where all the elements are value. More...
class	tvm::tir::Broadcast
	Managed reference to BroadcastNode. More...
class	tvm::tir::LetNode
	Let binding. Bind var to value then evaluate body. More...
class	tvm::tir::Let
	Managed reference to LetNode. More...
class	tvm::tir::CallNode
	Call node. More...
class	tvm::tir::Call
	Managed reference to CallNode. More...
class	tvm::tir::ShuffleNode
	Shuffle instruction. vec = concat(vectors) result = (vec[indices[0]], vec[indices[1]] ...) More...
class	tvm::tir::Shuffle
	Managed reference to ShuffleNode. More...
class	tvm::tir::CommReducerNode
	A commutative reducer node to represent a commutative binary operator with identity element. More...
class	tvm::tir::CommReducer
	Managed reference to CommReducerNode. More...
class	tvm::tir::ReduceNode
	Reduction operator. More...
class	tvm::tir::Reduce
	Managed reference to ReduceNode. More...
class	tvm::tir::AnyNode
	Any shape. More...
class	tvm::tir::Any
	Managed reference to AnyNode. More...
struct	tvm::runtime::PackedFuncValueConverter< tvm::tir::StringImm >
struct	tvm::runtime::PackedFuncValueConverter< PrimExpr >

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::tir
	tvm::runtime

Typedefs
using	tvm::tir::IntImmNode = tvm::IntImmNode
using	tvm::tir::FloatImmNode = tvm::FloatImmNode

Functions
template<typename K , typename V >
std::unordered_map< K, V >	tvm::tir::as_unordered_map (const Map< K, V > &dmap)

Detailed Description

TIR expressions.