tvm::script::ir_builder::tir::axis Namespace Reference

Functions
Var	Spatial (Range dom, PrimExpr binding, DataType dtype=DataType::Int(32))
	The spatial block axis defining function. More...
Var	Reduce (Range dom, PrimExpr binding, DataType dtype=DataType::Int(32))
	The reduced block axis defining function. More...
Var	Scan (Range dom, PrimExpr binding, DataType dtype=DataType::Int(32))
	The scanning block axis defining function. More...
Var	Opaque (Range dom, PrimExpr binding, DataType dtype=DataType::Int(32))
	The opaque block axis defining function. More...
Array< Var >	Remap (String kinds, Array< PrimExpr > bindings, DataType dtype=DataType::Int(32))
	The block axis remapping function. More...

Function Documentation

Opaque()

Var tvm::script::ir_builder::tir::axis::Opaque	(	Range	dom,
		PrimExpr	binding,
		DataType	dtype = DataType::Int(32)
	)

The opaque block axis defining function.

Parameters

dom	The domain of the iteration variable.
binding	The binding value of the iteration variable.
dtype	The data type of the iteration variable.

Returns

The iteration variable.

Reduce()

Var tvm::script::ir_builder::tir::axis::Reduce	(	Range	dom,
		PrimExpr	binding,
		DataType	dtype = DataType::Int(32)
	)

The reduced block axis defining function.

Parameters

dom	The domain of the iteration variable.
binding	The binding value of the iteration variable.
dtype	The data type of the iteration variable.

Returns

The iteration variable.

Remap()

Array<Var> tvm::script::ir_builder::tir::axis::Remap	(	String	kinds,
		Array< PrimExpr >	bindings,
		DataType	dtype = DataType::Int(32)
	)

The block axis remapping function.

Parameters

kinds	The types of the iteration variables.
bindings	The binding values of the iteration variables.
dtype	The data types of the iteration variables.

Returns

The iteration variables.

Scan()

Var tvm::script::ir_builder::tir::axis::Scan	(	Range	dom,
		PrimExpr	binding,
		DataType	dtype = DataType::Int(32)
	)

The scanning block axis defining function.

Parameters

dom	The domain of the iteration variable.
binding	The binding value of the iteration variable.
dtype	The data type of the iteration variable.

Returns

The iteration variable.

Spatial()

Var tvm::script::ir_builder::tir::axis::Spatial	(	Range	dom,
		PrimExpr	binding,
		DataType	dtype = DataType::Int(32)
	)

The spatial block axis defining function.

Parameters

dom	The domain of the iteration variable.
binding	The binding value of the iteration variable.
dtype	The data type of the iteration variable.

Returns

The iteration variable.