Tensor expression language DSL. More...

Classes
class	BaseComputeOpNode
	A Compute op that compute a tensor on certain domain. This is the base class for ComputeOp (operating on a scalar at a time) More...

class	CommReducer
	Managed reference to CommReducerNode. More...

class	CommReducerNode
	A commutative reducer node to represent a commutative binary operator with identity element. More...

class	ComputeOp
	Managed reference to ComputeOpNode. More...

class	ComputeOpNode
	A Compute op that compute a tensor on certain domain. More...

class	ExternOp
	Managed reference to ExternOpNode. More...

class	ExternOpNode
	External computation that cannot be splitted. More...

class	Operation
	Operation that produces tensors. More...

class	OperationNode
	Base class of all operation nodes. More...

class	PlaceholderOp
	Managed reference to PlaceholderOpNode. More...

class	PlaceholderOpNode
	A placeholder op represents an input placeholder. More...

class	Reduce
	Managed reference to ReduceNode. More...

class	ReduceNode
	Reduction operator. More...

class	ScanOp
	Managed reference to ScanOpNode. More...

class	ScanOpNode
	Symbolic scan. More...

class	Tensor
	Tensor structure representing a possible input, or intermediate computation result. More...

struct	TensorDom
	Temporary data structure to store union of bounds of each axis of Tensor. More...

class	TensorNode
	Opaque construction-time node that represents a tensor. More...

Typedefs
using	FCompute = std::function< PrimExpr(const ffi::Array< PrimVar > &i)>
	The compute function to specify the input source of a Tensor.

using	FBatchCompute = std::function< ffi::Array< PrimExpr >(const ffi::Array< PrimVar > &i)>
	The compute function to specify the inputs source of Tensors.

Functions
PrimVar	var (std::string name_hint, PrimType t=PrimType::Int(32))
	Construct a new Var expression.

IterVar	thread_axis (Range dom, std::string tag)
	Create a new IterVar that represents an axis in thread.

IterVar	reduce_axis (Range dom, std::string name="rv")
	Create a new IterVar for reduction operations.

Tensor	placeholder (ffi::Array< PrimExpr > shape, PrimType dtype=PrimType::Float(32), std::string name="placeholder")
	create a place holder tensor.

Tensor	compute (ffi::Array< PrimExpr > shape, FCompute fcompute, std::string name="tensor", std::string tag="", ffi::Map< ffi::String, ffi::Any > attrs={})
	Construct a new tensor by computing over shape, using the computation rule: result_tensor[axis] = fcompute(axis)

ffi::Array< Tensor >	compute (ffi::Array< PrimExpr > shape, FBatchCompute fcompute, std::string name="tensor", std::string tag="", ffi::Map< ffi::String, ffi::Any > attrs={})
	Construct a new tensor by computing over shape, using the computation rule: result_tensor[axis] = fcompute(axis)

ffi::Array< Tensor >	scan (ffi::Array< Tensor > init, ffi::Array< Tensor > update, ffi::Array< Tensor > state_placeholder, ffi::Array< Tensor > inputs=ffi::Array< Tensor >(), std::string name="scan", std::string tag="", ffi::Map< ffi::String, ffi::Any > attrs={})
	Construct new tensors by scan.

Tensor	compute (ffi::Array< PrimExpr > shape, std::function< PrimExpr(PrimVar)> f, std::string name="tensor", std::string tag="", ffi::Map< ffi::String, ffi::Any > attrs={})

Tensor	compute (ffi::Array< PrimExpr > shape, std::function< PrimExpr(PrimVar, PrimVar)> f, std::string name="tensor", std::string tag="", ffi::Map< ffi::String, ffi::Any > attrs={})

Tensor	compute (ffi::Array< PrimExpr > shape, std::function< PrimExpr(PrimVar, PrimVar, PrimVar)> f, std::string name="tensor", std::string tag="", ffi::Map< ffi::String, ffi::Any > attrs={})

Tensor	compute (ffi::Array< PrimExpr > shape, std::function< PrimExpr(PrimVar, PrimVar, PrimVar, PrimVar)> f, std::string name="tensor", std::string tag="", ffi::Map< ffi::String, ffi::Any > attrs={})

bool	IsTensorLoad (const Expr &expr)
	Return whether an expression is a Call whose callee is a TE Tensor.

Tensor	GetTensorFromLoad (const Call &call)
	Recover and validate the Tensor callee of a tensor-load Call.

ffi::Array< PrimExpr >	GetTensorLoadIndices (const Call &call)
	Recover and validate the primitive indices of a tensor-load Call.

PrimExpr	operator! (const Tensor::Slice &a)

PrimExpr	operator- (const Tensor::Slice &a)

template<typename T >
PrimExpr	operator+ (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator+ (const T &a, const Tensor::Slice &b)

PrimExpr	operator+ (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator- (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator- (const T &a, const Tensor::Slice &b)

PrimExpr	operator- (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator* (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator* (const T &a, const Tensor::Slice &b)

PrimExpr	operator* (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator== (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator== (const T &a, const Tensor::Slice &b)

PrimExpr	operator== (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator<= (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator<= (const T &a, const Tensor::Slice &b)

PrimExpr	operator<= (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator>= (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator>= (const T &a, const Tensor::Slice &b)

PrimExpr	operator>= (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator!= (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator!= (const T &a, const Tensor::Slice &b)

PrimExpr	operator!= (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator&& (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator&& (const T &a, const Tensor::Slice &b)

PrimExpr	operator&& (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator\|\| (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator\|\| (const T &a, const Tensor::Slice &b)

PrimExpr	operator\|\| (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator>> (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator>> (const T &a, const Tensor::Slice &b)

PrimExpr	operator>> (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator<< (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator<< (const T &a, const Tensor::Slice &b)

PrimExpr	operator<< (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator> (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator> (const T &a, const Tensor::Slice &b)

PrimExpr	operator> (const Tensor::Slice &a, const Tensor::Slice &b)

template<typename T >
PrimExpr	operator< (const Tensor::Slice &a, const T &b)

template<typename T >
PrimExpr	operator< (const T &a, const Tensor::Slice &b)

PrimExpr	operator< (const Tensor::Slice &a, const Tensor::Slice &b)

Detailed Description

Tensor expression language DSL.

Typedef Documentation

◆ FBatchCompute

using tvm::te::FBatchCompute = typedef std::function<ffi::Array<PrimExpr>(const ffi::Array<PrimVar>& i)>

The compute function to specify the inputs source of Tensors.

◆ FCompute

using tvm::te::FCompute = typedef std::function<PrimExpr(const ffi::Array<PrimVar>& i)>

The compute function to specify the input source of a Tensor.

Function Documentation

◆ compute() [1/6]

ffi::Array< Tensor > tvm::te::compute	(	ffi::Array< PrimExpr >	shape,
		FBatchCompute	fcompute,
		std::string	name = `"tensor"`,
		std::string	tag = `""`,
		ffi::Map< ffi::String, ffi::Any >	attrs = `{}`
	)

Construct a new tensor by computing over shape, using the computation rule: result_tensor[axis] = fcompute(axis)

Parameters

shape	Shape of the tensor.
fcompute	The compute function to create the tensors.
name	The optional name of the tensor.
tag	The optional tag of the tensor.
attrs	Optional additional attributes of the compute.

◆ compute() [2/6]

Tensor tvm::te::compute	(	ffi::Array< PrimExpr >	shape,
		FCompute	fcompute,
		std::string	name = `"tensor"`,
		std::string	tag = `""`,
		ffi::Map< ffi::String, ffi::Any >	attrs = `{}`
	)

Construct a new tensor by computing over shape, using the computation rule: result_tensor[axis] = fcompute(axis)

Parameters

shape	Shape of the tensor.
fcompute	The compute function to create the tensor.
name	The optional name of the tensor.
tag	The optional tag of the tensor.
attrs	Optional additional attributes of the compute.

◆ compute() [3/6]

Tensor tvm::te::compute	(	ffi::Array< PrimExpr >	shape,
		std::function< PrimExpr(PrimVar)>	f,
		std::string	name = `"tensor"`,
		std::string	tag = `""`,
		ffi::Map< ffi::String, ffi::Any >	attrs = `{}`
	)

inline

◆ compute() [4/6]

Tensor tvm::te::compute	(	ffi::Array< PrimExpr >	shape,
		std::function< PrimExpr(PrimVar, PrimVar)>	f,
		std::string	name = `"tensor"`,
		std::string	tag = `""`,
		ffi::Map< ffi::String, ffi::Any >	attrs = `{}`
	)

inline

◆ compute() [5/6]

Tensor tvm::te::compute	(	ffi::Array< PrimExpr >	shape,
		std::function< PrimExpr(PrimVar, PrimVar, PrimVar)>	f,
		std::string	name = `"tensor"`,
		std::string	tag = `""`,
		ffi::Map< ffi::String, ffi::Any >	attrs = `{}`
	)

inline

◆ compute() [6/6]

Tensor tvm::te::compute	(	ffi::Array< PrimExpr >	shape,
		std::function< PrimExpr(PrimVar, PrimVar, PrimVar, PrimVar)>	f,
		std::string	name = `"tensor"`,
		std::string	tag = `""`,
		ffi::Map< ffi::String, ffi::Any >	attrs = `{}`
	)

inline

◆ GetTensorFromLoad()

Tensor tvm::te::GetTensorFromLoad ( const Call & call )

Recover and validate the Tensor callee of a tensor-load Call.

◆ GetTensorLoadIndices()

ffi::Array< PrimExpr > tvm::te::GetTensorLoadIndices ( const Call & call )

Recover and validate the primitive indices of a tensor-load Call.

◆ IsTensorLoad()

bool tvm::te::IsTensorLoad ( const Expr & expr )

Return whether an expression is a Call whose callee is a TE Tensor.

◆ operator!()

PrimExpr tvm::te::operator! ( const Tensor::Slice & a )

inline

◆ operator!=() [1/3]

template<typename T >

PrimExpr tvm::te::operator!=	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator!=() [2/3]

template<typename T >

PrimExpr tvm::te::operator!=	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator!=() [3/3]

PrimExpr tvm::te::operator!=	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator&&() [1/3]

template<typename T >

PrimExpr tvm::te::operator&&	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator&&() [2/3]

template<typename T >

PrimExpr tvm::te::operator&&	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator&&() [3/3]

PrimExpr tvm::te::operator&&	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator*() [1/3]

template<typename T >

PrimExpr tvm::te::operator*	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator*() [2/3]

template<typename T >

PrimExpr tvm::te::operator*	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator*() [3/3]

PrimExpr tvm::te::operator*	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator+() [1/3]

template<typename T >

PrimExpr tvm::te::operator+	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator+() [2/3]

template<typename T >

PrimExpr tvm::te::operator+	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator+() [3/3]

PrimExpr tvm::te::operator+	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator-() [1/4]

template<typename T >

PrimExpr tvm::te::operator-	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator-() [2/4]

PrimExpr tvm::te::operator- ( const Tensor::Slice & a )

inline

◆ operator-() [3/4]

template<typename T >

PrimExpr tvm::te::operator-	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator-() [4/4]

PrimExpr tvm::te::operator-	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator<() [1/3]

template<typename T >

PrimExpr tvm::te::operator<	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator<() [2/3]

template<typename T >

PrimExpr tvm::te::operator<	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator<() [3/3]

PrimExpr tvm::te::operator<	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator<<() [1/3]

template<typename T >

PrimExpr tvm::te::operator<<	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator<<() [2/3]

template<typename T >

PrimExpr tvm::te::operator<<	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator<<() [3/3]

PrimExpr tvm::te::operator<<	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator<=() [1/3]

template<typename T >

PrimExpr tvm::te::operator<=	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator<=() [2/3]

template<typename T >

PrimExpr tvm::te::operator<=	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator<=() [3/3]

PrimExpr tvm::te::operator<=	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator==() [1/3]

template<typename T >

PrimExpr tvm::te::operator==	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator==() [2/3]

template<typename T >

PrimExpr tvm::te::operator==	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator==() [3/3]

PrimExpr tvm::te::operator==	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator>() [1/3]

template<typename T >

PrimExpr tvm::te::operator>	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator>() [2/3]

template<typename T >

PrimExpr tvm::te::operator>	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator>() [3/3]

PrimExpr tvm::te::operator>	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator>=() [1/3]

template<typename T >

PrimExpr tvm::te::operator>=	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator>=() [2/3]

template<typename T >

PrimExpr tvm::te::operator>=	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator>=() [3/3]

PrimExpr tvm::te::operator>=	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator>>() [1/3]

template<typename T >

PrimExpr tvm::te::operator>>	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator>>() [2/3]

template<typename T >

PrimExpr tvm::te::operator>>	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator>>() [3/3]

PrimExpr tvm::te::operator>>	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator||() [1/3]

template<typename T >

PrimExpr tvm::te::operator\|\|	(	const T &	a,
		const Tensor::Slice &	b
	)

inline

◆ operator||() [2/3]

template<typename T >

PrimExpr tvm::te::operator\|\|	(	const Tensor::Slice &	a,
		const T &	b
	)

inline

◆ operator||() [3/3]

PrimExpr tvm::te::operator\|\|	(	const Tensor::Slice &	a,
		const Tensor::Slice &	b
	)

inline

◆ placeholder()

Tensor tvm::te::placeholder	(	ffi::Array< PrimExpr >	shape,
		PrimType	dtype = `PrimType::Float(32)`,
		std::string	name = `"placeholder"`
	)

create a place holder tensor.

Parameters

shape	The shape of the tensor.
dtype	the data type of the tensor.
name	The name of the Tensor.

◆ reduce_axis()

IterVar tvm::te::reduce_axis	(	Range	dom,
		std::string	name = `"rv"`
	)

Create a new IterVar for reduction operations.

Parameters

dom	The domain of the reduction axis.
name	The name of the reduction axis.

◆ scan()

ffi::Array< Tensor > tvm::te::scan	(	ffi::Array< Tensor >	init,
		ffi::Array< Tensor >	update,
		ffi::Array< Tensor >	state_placeholder,
		ffi::Array< Tensor >	inputs = `ffi::Array< Tensor >()`,
		std::string	name = `"scan"`,
		std::string	tag = `""`,
		ffi::Map< ffi::String, ffi::Any >	attrs = `{}`
	)

Construct new tensors by scan.

Parameters

init	The intialize tensor of first K steps.
update	The update tensor indicated the updated result after each timestamp.
state_placeholder	The placeholder for the states.
inputs	The inputs to the scan body, this is optional, but recommended to provide concrete information about scan body.
name	The optional name of the tensor.
tag	The optional tag of the tensor.
attrs	Optional additional attributes of the compute.

◆ thread_axis()

IterVar tvm::te::thread_axis	(	Range	dom,
		std::string	tag
	)

Create a new IterVar that represents an axis in thread.

Parameters

dom	Optional, domain of the thread axis.
tag	The thread tag of the axis.

◆ var()

PrimVar tvm::te::var	(	std::string	name_hint,
		PrimType	t = `PrimType::Int(32)`
	)

Construct a new Var expression.

Parameters

name_hint	The name hint for the expression
t	The type of the expression

Classes

Typedefs

Functions

Detailed Description

Typedef Documentation

◆ FBatchCompute

◆ FCompute

Function Documentation

◆ compute() [1/6]

◆ compute() [2/6]

◆ compute() [3/6]

◆ compute() [4/6]

◆ compute() [5/6]

◆ compute() [6/6]

◆ GetTensorFromLoad()

◆ GetTensorLoadIndices()

◆ IsTensorLoad()

◆ operator!()

◆ operator!=() [1/3]

◆ operator!=() [2/3]

◆ operator!=() [3/3]

◆ operator&&() [1/3]

◆ operator&&() [2/3]

◆ operator&&() [3/3]

◆ operator*() [1/3]

◆ operator*() [2/3]

◆ operator*() [3/3]

◆ operator+() [1/3]

◆ operator+() [2/3]

◆ operator+() [3/3]

◆ operator-() [1/4]

◆ operator-() [2/4]

◆ operator-() [3/4]

◆ operator-() [4/4]

◆ operator<() [1/3]

◆ operator<() [2/3]

◆ operator<() [3/3]

◆ operator<<() [1/3]

◆ operator<<() [2/3]

◆ operator<<() [3/3]

◆ operator<=() [1/3]

◆ operator<=() [2/3]

◆ operator<=() [3/3]

◆ operator==() [1/3]

◆ operator==() [2/3]

◆ operator==() [3/3]

◆ operator>() [1/3]

◆ operator>() [2/3]

◆ operator>() [3/3]

◆ operator>=() [1/3]

◆ operator>=() [2/3]

◆ operator>=() [3/3]

◆ operator>>() [1/3]

◆ operator>>() [2/3]

◆ operator>>() [3/3]

◆ operator||() [1/3]

◆ operator||() [2/3]

◆ operator||() [3/3]

◆ placeholder()

◆ reduce_axis()

◆ scan()

◆ thread_axis()

◆ var()