function.h File Reference

TIR Function. More...

#include <tvm/ir/function.h>
#include <tvm/runtime/ndarray.h>
#include <tvm/tir/buffer.h>
#include <tvm/tir/expr.h>
#include <tvm/tir/stmt.h>
#include <string>

Include dependency graph for function.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::tir::PrimFuncNode
	Primitive functions that contains TIR statements. More...
class	tvm::tir::PrimFunc
	Managed reference to PrimFuncNode. More...
class	tvm::tir::LinkedParamNode
	Describes one parameter that should be linked into the generated module. More...
class	tvm::tir::LinkedParam
	Managed reference to LinkedParamNode. More...

Namespaces
	tvm
	Performance counters for profiling via the PAPI library.
	tvm::tir
	tvm::tir::attr
	PrimFunc specific attribute names.

Functions
PrimFunc	tvm::tir::Specialize (PrimFunc func, const Map< Var, ObjectRef > &param_map)
	Specialize parameters of PrimFunc. More...

Variables
constexpr const char *	tvm::tir::attr::kDeviceThreadAxis = "tir.device_thread_axis"
	List of thread IterVar that a DeviceLaunch function corresponds to. More...
constexpr const char *	tvm::tir::attr::kDeviceUseDynSharedMemory = "tir.device_use_dyn_shared_memory"
	Whether or not use dynamic shared memory. More...
constexpr const char *	tvm::tir::attr::kNoAlias = "tir.noalias"
	Whether to set noalias rule on the function arguments. More...
constexpr const char *	tvm::tir::attr::kIsEntryFunc = "tir.is_entry_func"
	Mark the function as the entry function of the final generated runtime module. More...
constexpr const char *	tvm::tir::attr::kLinkedParams = "tir.linked_params"
	Parameters used in the module that should be linked by the codegen. More...

Detailed Description

TIR Function.