FreeTensor
Loading...
Searching...
No Matches
Public Member Functions | List of all members
freetensor::FrontendVar Class Reference

#include <frontend_var.h>

Public Member Functions

 FrontendVar (const std::string &name, const std::vector< Expr > &fullShape, DataType dtype, MemType mtype, const std::vector< FrontendVarIdx > &indices, bool isLoadAtVersion=false)
 
const std::string & name () const
 
const std::vector< Expr > & fullShape () const
 
DataType dtype () const
 
MemType mtype () const
 
int ndim () const
 
const std::vector< FrontendVarIdx > & indices () const
 
Expr asLoad () const
 
Stmt asStore (const Metadata &metadata, const Expr &value) const
 
Stmt asReduceTo (ReduceOp op, const Metadata &metadata, const Expr &value, bool atomic=false) const
 
std::vector< FrontendVarIdxchainIndices (const std::vector< FrontendVarIdx > &next) const
 
Expr shape (const Expr &idx) const
 
std::vector< Exprshape () const
 

Constructor & Destructor Documentation

◆ FrontendVar()

freetensor::FrontendVar::FrontendVar ( const std::string &  name,
const std::vector< Expr > &  fullShape,
DataType  dtype,
MemType  mtype,
const std::vector< FrontendVarIdx > &  indices,
bool  isLoadAtVersion = false 
)
inline

Member Function Documentation

◆ asLoad()

Expr freetensor::FrontendVar::asLoad ( ) const

◆ asReduceTo()

Stmt freetensor::FrontendVar::asReduceTo ( ReduceOp  op,
const Metadata metadata,
const Expr value,
bool  atomic = false 
) const

◆ asStore()

Stmt freetensor::FrontendVar::asStore ( const Metadata metadata,
const Expr value 
) const

◆ chainIndices()

std::vector< FrontendVarIdx > freetensor::FrontendVar::chainIndices ( const std::vector< FrontendVarIdx > &  next) const

◆ dtype()

DataType freetensor::FrontendVar::dtype ( ) const
inline

◆ fullShape()

const std::vector< Expr > & freetensor::FrontendVar::fullShape ( ) const
inline

shape of the original variable before slicing

◆ indices()

const std::vector< FrontendVarIdx > & freetensor::FrontendVar::indices ( ) const
inline

◆ mtype()

MemType freetensor::FrontendVar::mtype ( ) const
inline

◆ name()

const std::string & freetensor::FrontendVar::name ( ) const
inline

◆ ndim()

int freetensor::FrontendVar::ndim ( ) const

Number of dimensions after slicing

◆ shape() [1/2]

std::vector< Expr > freetensor::FrontendVar::shape ( ) const

◆ shape() [2/2]

Expr freetensor::FrontendVar::shape ( const Expr idx) const

shape of each dimension after slicing


The documentation for this class was generated from the following files: