expr.h File Reference

TIRx tensor expressions and construction helpers. More...

#include <tvm/ffi/container/array.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ffi/string.h>
#include <tvm/ir/expr.h>
#include <tvm/tirx/op/memory.h>
#include <tvm/tirx/type.h>
#include <string>

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::tirx::TensorVar
	Checked zero-state view over an ordinary VarNode with TensorType. More...
struct	tvm::ffi::TypeTraits< tirx::TensorVar >

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::tirx
	Backend-defined execution hierarchy nodes shared by layouts and tile calls.
namespace	tvm::ffi

Functions
bool	tvm::tirx::operator== (const TensorVar &lhs, const TensorVar &rhs)
bool	tvm::tirx::operator!= (const TensorVar &lhs, const TensorVar &rhs)
TensorVar	tvm::tirx::GetTensorVar (const VarNode *var)
	Recover a checked tensor view from an ordinary VarNode pointer.
ffi::ObjectPtr< TensorTypeNode >	tvm::tirx::CopyTensorType (const TensorVar &var)
TensorVar	tvm::tirx::RebuildTensorVar (const TensorVar &var, ffi::ObjectPtr< TensorTypeNode > type, ffi::Optional< ffi::String > name=std::nullopt)
TVM_TIRX_DLL TensorVar	tvm::tirx::decl_tensor (ffi::Array< PrimExpr > shape, PrimType dtype=PrimType::Float(32), ffi::String name="tensor", ffi::String storage_scope="", Location loc=UnknownLoc())
	Construct a new tensor given shape, and dtype.
TVM_TIRX_DLL tirx::TensorVar	tvm::tirx::TensorWithOffsetAlignment (ffi::Array< PrimExpr > shape, PrimType dtype, std::string name, int data_alignment, int offset_factor, std::string memory_scope="")
	Creates a TIR tensor for the provided parameters.
TVM_TIRX_DLL TensorLoad	tvm::tirx::MakeTensorLoad (TensorVar tensor, ffi::Array< PrimExpr > indices, Location loc=UnknownLoc())
	Construct a TensorLoad from a TensorVar.
TVM_TIRX_DLL TensorRegion	tvm::tirx::FullTensorRegion (TensorVar tensor)
	Select the entire tensor.
TVM_TIRX_DLL TensorRegion	tvm::tirx::TensorRegionFromPoint (TensorVar tensor, ffi::Array< PrimExpr > indices)
	Construct unit or vector-lane ranges from point indices.

Detailed Description

TIRx tensor expressions and construction helpers.