expr.h File Reference

#include <tvm/ffi/container/array.h>
#include <tvm/ffi/container/map.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ir/cow.h>
#include <tvm/ir/expr.h>
#include <tvm/ir/function.h>
#include <tvm/ir/location.h>
#include <tvm/ir/prim/expr.h>
#include <tvm/ir/prim/op.h>
#include <tvm/relax/type.h>
#include <tvm/runtime/tensor.h>
#include <functional>

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::relax::ShapeExprNode
	A shape expression which allows users to construct a shape containing PrimExpr. More...
class	tvm::relax::ShapeExpr
class	tvm::relax::DataflowVarNode
	A sub-type of the variable node used to mark dataflow variables from normal visible "function local" bindings. More...
class	tvm::relax::DataflowVar
class	tvm::relax::BindingNode
	The base class of a variable binding in Relax. More...
class	tvm::relax::Binding
class	tvm::relax::MatchCastNode
	Runtime-match the value to the type. More...
class	tvm::relax::MatchCast
	Managed reference to MatchCastNode. More...
class	tvm::relax::VarBindingNode
class	tvm::relax::VarBinding
class	tvm::relax::BindingBlockNode
class	tvm::relax::BindingBlock
class	tvm::relax::DataflowBlockNode
class	tvm::relax::DataflowBlock
class	tvm::relax::SeqExprNode
	A sequence of blocks followed by an expression. More...
class	tvm::relax::SeqExpr
class	tvm::relax::IfExprNode
	Condition expression. More...
class	tvm::relax::IfExpr
class	tvm::relax::FunctionNode
	A Relax function. More...
class	tvm::relax::Function
class	tvm::relax::ExternFuncNode
	The extern function, which can represent packed function. More...
class	tvm::relax::ExternFunc

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

Functions
GenericConst	tvm::relax::MakeTensorConst (ffi::Tensor data, ffi::Optional< Type > ty_annotation=std::nullopt, Location loc=UnknownLoc())
	Construct a tensor constant, inferring shape and dtype when type is omitted.
Expr	tvm::relax::GetShapeOf (const Expr &expr)
	Get the shape of Expr.

Variables
constexpr const char *	tvm::relax::attr::kPrimitive = "Primitive"
	Mark the function as a primitive function.
constexpr const char *	tvm::relax::attr::kCodegen = "Codegen"
	Indicate the codegen that should be used for building this function. When this is unset or set to "default", the default compilation pipeline will be used.
constexpr const char *	tvm::relax::attr::kComposite = "Composite"
	Treat the function as a composite operator.
constexpr const char *	tvm::relax::attr::kPartitionedFromPattern = "PartitionedFromPattern"
	Indicate the function was created by the Pattern Partitioning Pass.
constexpr const char *	tvm::relax::attr::kWorkspaceSize = "WorkspaceSize"
	The required workspace for an external function.
constexpr const char *	tvm::relax::attr::kForcePure = "relax.force_pure"
	Override checking purity for this function and treat as pure (is_pure must be set to true)
constexpr const char *	tvm::relax::attr::kNumInput = "num_input"
	The number of inputs of a function. If a function has the num_input attribute, the last func->params.size() - num_inputs arguments are assumed to be weights that are fixed across invocations.
constexpr const char *	tvm::relax::attr::kOpPattern = "op_pattern"
constexpr const char *	tvm::relax::attr::kOpAttrs = "op_attrs"
constexpr const char *	tvm::relax::attr::kOperatorName = "operator_name"
constexpr const char *	tvm::relax::attr::kLibraryKernel = "library_kernel"
constexpr const char *	tvm::relax::attr::kCSource = "c_source"
constexpr const char *	tvm::relax::attr::kCSourceFmt = "c_source_fmt"
constexpr const char *	tvm::relax::attr::kTirVarUpperBound = "tir_var_upper_bound"
constexpr const char *	tvm::relax::attr::kTirVarLowerBound = "tir_var_lower_bound"
constexpr const char *	tvm::relax::attr::kTirNonNegativeVar = "tir_non_negative_var"
constexpr const char *	tvm::relax::attr::kMemoryPlanDynamicFuncOutput = "relax.memory_plan_dynamic_func_output"
constexpr const char *	tvm::relax::attr::kCaptureSymbolicVars = "relax.rewrite_cuda_graph.capture_symbolic_vars"
constexpr const char *	tvm::relax::attr::kScoped = "scoped"