trn.h File Reference

#include <tvm/tirx/expr.h>
#include <tvm/tirx/op.h>

Include dependency graph for trn.h:

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

Go to the source code of this file.

Namespaces
	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
	tvm::tirx
	tvm::tirx::builtin
	Collection of builtin intrinsics as ops.

Functions
const Op &	tvm::tirx::builtin::nki_load ()
	nki intrinsics for load operation. More...
const Op &	tvm::tirx::builtin::nki_store ()
	nki intrinsics for store operation. More...
const Op &	tvm::tirx::builtin::nki_tensor_copy ()
	nki intrinsics for tensor_copy operation. More...
const Op &	tvm::tirx::builtin::nki_matmul ()
	nki intrinsics for matmul operation. More...
const Op &	tvm::tirx::builtin::nki_activation ()
	nki intrinsics for activation operation. More...
const Op &	tvm::tirx::builtin::nki_reciprocal ()
	nki intrinsics for reciprocal operation. More...
const Op &	tvm::tirx::builtin::nki_tensortensor ()
	nki intrinsics for tensortensor operation. More...
const Op &	tvm::tirx::builtin::nki_tensorscalar ()
	nki intrinsics for tensorscalar operation. More...
const Op &	tvm::tirx::builtin::nki_tensorreduce ()
	nki intrinsics for tensorreduce operation. More...
const Op &	tvm::tirx::builtin::nki_memset ()
	nki intrinsics for memset operation. More...
const Op &	tvm::tirx::builtin::nki_activation_reduce ()
	nki intrinsics for activation reduce operation. More...
const Op &	tvm::tirx::builtin::nki_tensorscalar_reduce ()
	nki intrinsics for tensorscalar reduce operation. More...
const Op &	tvm::tirx::builtin::nki_identity ()
	nki intrinsics for initializing identity tensor. More...
const Op &	tvm::tirx::builtin::nki_scalar_tensor_tensor ()
	nki intrinsics for scalar tensor tensor operation. More...
const Op &	tvm::tirx::builtin::nki_scalar_tensor_scalar ()
	nki intrinsics for scalar tensor scalar operation. More...
const Op &	tvm::tirx::builtin::nki_affine_select ()
	nki intrinsics for affine_select operation. More...