TT-MLIR
Public Types | Public Member Functions | Static Public Member Functions | Private Member Functions | Static Private Member Functions | List of all members
mlir::tt::ttnn::ArangeOp Class Reference
Inheritance diagram for mlir::tt::ttnn::ArangeOp:

Public Types

using Adaptor = ArangeOpAdaptor
 
template<typename RangeT >
using GenericAdaptor = ArangeOpGenericAdaptor< RangeT >
 
using FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > >
 
using Properties = FoldAdaptor::Properties
 

Public Member Functions

static ::llvm::ArrayRef<::llvm::StringRef > getAttributeNames ()
 
::mlir::StringAttr getDtypeAttrName ()
 
::mlir::StringAttr getEndAttrName ()
 
::mlir::StringAttr getMemoryConfigAttrName ()
 
::mlir::StringAttr getStartAttrName ()
 
::mlir::StringAttr getStepAttrName ()
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index)
 
::mlir::Operation::operand_range getODSOperands (unsigned index)
 
::mlir::TypedValue<::mlir::tt::DeviceTypegetDevice ()
 
::mlir::MutableOperandRange getDeviceMutable ()
 
std::pair< unsigned, unsigned > getODSResultIndexAndLength (unsigned index)
 
::mlir::Operation::result_range getODSResults (unsigned index)
 
::mlir::TypedValue<::mlir::RankedTensorType > getResult ()
 
void writeProperties (::mlir::DialectBytecodeWriter &writer)
 
::mlir::IntegerAttr getStartAttr ()
 
uint64_t getStart ()
 
::mlir::IntegerAttr getEndAttr ()
 
uint64_t getEnd ()
 
::mlir::IntegerAttr getStepAttr ()
 
uint64_t getStep ()
 
::mlir::tt::DataTypeAttr getDtypeAttr ()
 
::std::optional<::mlir::tt::DataTypegetDtype ()
 
::mlir::tt::ttnn::MemoryConfigAttr getMemoryConfigAttr ()
 
::std::optional<::mlir::tt::ttnn::MemoryConfigAttrgetMemoryConfig ()
 
void setStartAttr (::mlir::IntegerAttr attr)
 
void setStart (uint64_t attrValue)
 
void setEndAttr (::mlir::IntegerAttr attr)
 
void setEnd (uint64_t attrValue)
 
void setStepAttr (::mlir::IntegerAttr attr)
 
void setStep (uint64_t attrValue)
 
void setDtypeAttr (::mlir::tt::DataTypeAttr attr)
 
void setDtype (::std::optional<::mlir::tt::DataType > attrValue)
 
void setMemoryConfigAttr (::mlir::tt::ttnn::MemoryConfigAttr attr)
 
::mlir::Attribute removeDtypeAttr ()
 
::mlir::Attribute removeMemoryConfigAttr ()
 
::llvm::LogicalResult verifyInvariantsImpl ()
 
::llvm::LogicalResult verifyInvariants ()
 
::llvm::LogicalResult verify ()
 

Static Public Member Functions

::mlir::StringAttr getDtypeAttrName (::mlir::OperationName name)
 
::mlir::StringAttr getEndAttrName (::mlir::OperationName name)
 
::mlir::StringAttr getMemoryConfigAttrName (::mlir::OperationName name)
 
::mlir::StringAttr getStartAttrName (::mlir::OperationName name)
 
::mlir::StringAttr getStepAttrName (::mlir::OperationName name)
 
static constexpr ::llvm::StringLiteral getOperationName ()
 
::llvm::LogicalResult setPropertiesFromAttr (Properties &prop, ::mlir::Attribute attr, ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError)
 
::mlir::Attribute getPropertiesAsAttr (::mlir::MLIRContext *ctx, const Properties &prop)
 
static llvm::hash_code computePropertiesHash (const Properties &prop)
 
static std::optional< mlir::Attribute > getInherentAttr (::mlir::MLIRContext *ctx, const Properties &prop, llvm::StringRef name)
 
static void setInherentAttr (Properties &prop, llvm::StringRef name, mlir::Attribute value)
 
static void populateInherentAttrs (::mlir::MLIRContext *ctx, const Properties &prop, ::mlir::NamedAttrList &attrs)
 
::llvm::LogicalResult verifyInherentAttrs (::mlir::OperationName opName, ::mlir::NamedAttrList &attrs, llvm::function_ref<::mlir::InFlightDiagnostic()> emitError)
 
::llvm::LogicalResult readProperties (::mlir::DialectBytecodeReader &reader, ::mlir::OperationState &state)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::IntegerAttr start, ::mlir::IntegerAttr end, ::mlir::IntegerAttr step, ::mlir::tt::DataTypeAttr dtype, ::mlir::Value device, ::mlir::tt::ttnn::MemoryConfigAttr memory_config)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::IntegerAttr start, ::mlir::IntegerAttr end, ::mlir::IntegerAttr step, ::mlir::tt::DataTypeAttr dtype, ::mlir::Value device, ::mlir::tt::ttnn::MemoryConfigAttr memory_config)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, uint64_t start, uint64_t end, uint64_t step, ::mlir::tt::DataTypeAttr dtype, ::mlir::Value device, ::mlir::tt::ttnn::MemoryConfigAttr memory_config)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, uint64_t start, uint64_t end, uint64_t step, ::mlir::tt::DataTypeAttr dtype, ::mlir::Value device, ::mlir::tt::ttnn::MemoryConfigAttr memory_config)
 
static void build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute > attributes={})
 

Private Member Functions

::mlir::StringAttr getAttributeNameForIndex (unsigned index)
 

Static Private Member Functions

::mlir::StringAttr getAttributeNameForIndex (::mlir::OperationName name, unsigned index)
 

Member Typedef Documentation

◆ Adaptor

◆ FoldAdaptor

using mlir::tt::ttnn::ArangeOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> >

◆ GenericAdaptor

template<typename RangeT >
using mlir::tt::ttnn::ArangeOp::GenericAdaptor = ArangeOpGenericAdaptor<RangeT>

◆ Properties

Member Function Documentation

◆ build() [1/5]

void mlir::tt::ttnn::ArangeOp::build ( ::mlir::OpBuilder &  ,
::mlir::OperationState &  odsState,
::mlir::TypeRange  resultTypes,
::mlir::ValueRange  operands,
::llvm::ArrayRef<::mlir::NamedAttribute >  attributes = {} 
)
static

◆ build() [2/5]

void mlir::tt::ttnn::ArangeOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::Type  result,
::mlir::IntegerAttr  start,
::mlir::IntegerAttr  end,
::mlir::IntegerAttr  step,
::mlir::tt::DataTypeAttr  dtype,
::mlir::Value  device,
::mlir::tt::ttnn::MemoryConfigAttr  memory_config 
)
static

◆ build() [3/5]

void mlir::tt::ttnn::ArangeOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::Type  result,
uint64_t  start,
uint64_t  end,
uint64_t  step,
::mlir::tt::DataTypeAttr  dtype,
::mlir::Value  device,
::mlir::tt::ttnn::MemoryConfigAttr  memory_config 
)
static

◆ build() [4/5]

void mlir::tt::ttnn::ArangeOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::TypeRange  resultTypes,
::mlir::IntegerAttr  start,
::mlir::IntegerAttr  end,
::mlir::IntegerAttr  step,
::mlir::tt::DataTypeAttr  dtype,
::mlir::Value  device,
::mlir::tt::ttnn::MemoryConfigAttr  memory_config 
)
static

◆ build() [5/5]

void mlir::tt::ttnn::ArangeOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::TypeRange  resultTypes,
uint64_t  start,
uint64_t  end,
uint64_t  step,
::mlir::tt::DataTypeAttr  dtype,
::mlir::Value  device,
::mlir::tt::ttnn::MemoryConfigAttr  memory_config 
)
static

◆ computePropertiesHash()

llvm::hash_code mlir::tt::ttnn::ArangeOp::computePropertiesHash ( const Properties prop)
static

◆ getAttributeNameForIndex() [1/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getAttributeNameForIndex ( ::mlir::OperationName  name,
unsigned  index 
)
inlinestaticprivate

◆ getAttributeNameForIndex() [2/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getAttributeNameForIndex ( unsigned  index)
inlineprivate

◆ getAttributeNames()

static ::llvm::ArrayRef<::llvm::StringRef> mlir::tt::ttnn::ArangeOp::getAttributeNames ( )
inline

◆ getDevice()

::mlir::TypedValue<::mlir::tt::DeviceType> mlir::tt::ttnn::ArangeOp::getDevice ( )
inline

◆ getDeviceMutable()

mlir::MutableOperandRange mlir::tt::ttnn::ArangeOp::getDeviceMutable ( )

◆ getDtype()

std::optional<::mlir::tt::DataType > mlir::tt::ttnn::ArangeOp::getDtype ( )

◆ getDtypeAttr()

::mlir::tt::DataTypeAttr mlir::tt::ttnn::ArangeOp::getDtypeAttr ( )
inline

◆ getDtypeAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getDtypeAttrName ( )
inline

◆ getDtypeAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getDtypeAttrName ( ::mlir::OperationName  name)
inlinestatic

◆ getEnd()

uint64_t mlir::tt::ttnn::ArangeOp::getEnd ( )

◆ getEndAttr()

::mlir::IntegerAttr mlir::tt::ttnn::ArangeOp::getEndAttr ( )
inline

◆ getEndAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getEndAttrName ( )
inline

◆ getEndAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getEndAttrName ( ::mlir::OperationName  name)
inlinestatic

◆ getInherentAttr()

std::optional< mlir::Attribute > mlir::tt::ttnn::ArangeOp::getInherentAttr ( ::mlir::MLIRContext *  ctx,
const Properties prop,
llvm::StringRef  name 
)
static

◆ getMemoryConfig()

std::optional<::mlir::tt::ttnn::MemoryConfigAttr > mlir::tt::ttnn::ArangeOp::getMemoryConfig ( )

◆ getMemoryConfigAttr()

::mlir::tt::ttnn::MemoryConfigAttr mlir::tt::ttnn::ArangeOp::getMemoryConfigAttr ( )
inline

◆ getMemoryConfigAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getMemoryConfigAttrName ( )
inline

◆ getMemoryConfigAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getMemoryConfigAttrName ( ::mlir::OperationName  name)
inlinestatic

◆ getODSOperandIndexAndLength()

std::pair< unsigned, unsigned > mlir::tt::ttnn::ArangeOp::getODSOperandIndexAndLength ( unsigned  index)

◆ getODSOperands()

::mlir::Operation::operand_range mlir::tt::ttnn::ArangeOp::getODSOperands ( unsigned  index)
inline

◆ getODSResultIndexAndLength()

std::pair<unsigned, unsigned> mlir::tt::ttnn::ArangeOp::getODSResultIndexAndLength ( unsigned  index)
inline

◆ getODSResults()

::mlir::Operation::result_range mlir::tt::ttnn::ArangeOp::getODSResults ( unsigned  index)
inline

◆ getOperationName()

static constexpr ::llvm::StringLiteral mlir::tt::ttnn::ArangeOp::getOperationName ( )
inlinestatic

◆ getPropertiesAsAttr()

mlir::Attribute mlir::tt::ttnn::ArangeOp::getPropertiesAsAttr ( ::mlir::MLIRContext *  ctx,
const Properties prop 
)
static

◆ getResult()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::ArangeOp::getResult ( )
inline

◆ getStart()

uint64_t mlir::tt::ttnn::ArangeOp::getStart ( )

◆ getStartAttr()

::mlir::IntegerAttr mlir::tt::ttnn::ArangeOp::getStartAttr ( )
inline

◆ getStartAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getStartAttrName ( )
inline

◆ getStartAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getStartAttrName ( ::mlir::OperationName  name)
inlinestatic

◆ getStep()

uint64_t mlir::tt::ttnn::ArangeOp::getStep ( )

◆ getStepAttr()

::mlir::IntegerAttr mlir::tt::ttnn::ArangeOp::getStepAttr ( )
inline

◆ getStepAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getStepAttrName ( )
inline

◆ getStepAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttnn::ArangeOp::getStepAttrName ( ::mlir::OperationName  name)
inlinestatic

◆ populateInherentAttrs()

void mlir::tt::ttnn::ArangeOp::populateInherentAttrs ( ::mlir::MLIRContext *  ctx,
const Properties prop,
::mlir::NamedAttrList &  attrs 
)
static

◆ readProperties()

llvm::LogicalResult mlir::tt::ttnn::ArangeOp::readProperties ( ::mlir::DialectBytecodeReader &  reader,
::mlir::OperationState &  state 
)
static

◆ removeDtypeAttr()

::mlir::Attribute mlir::tt::ttnn::ArangeOp::removeDtypeAttr ( )
inline

◆ removeMemoryConfigAttr()

::mlir::Attribute mlir::tt::ttnn::ArangeOp::removeMemoryConfigAttr ( )
inline

◆ setDtype()

void mlir::tt::ttnn::ArangeOp::setDtype ( ::std::optional<::mlir::tt::DataType attrValue)

◆ setDtypeAttr()

void mlir::tt::ttnn::ArangeOp::setDtypeAttr ( ::mlir::tt::DataTypeAttr  attr)
inline

◆ setEnd()

void mlir::tt::ttnn::ArangeOp::setEnd ( uint64_t  attrValue)

◆ setEndAttr()

void mlir::tt::ttnn::ArangeOp::setEndAttr ( ::mlir::IntegerAttr  attr)
inline

◆ setInherentAttr()

void mlir::tt::ttnn::ArangeOp::setInherentAttr ( Properties prop,
llvm::StringRef  name,
mlir::Attribute  value 
)
static

◆ setMemoryConfigAttr()

void mlir::tt::ttnn::ArangeOp::setMemoryConfigAttr ( ::mlir::tt::ttnn::MemoryConfigAttr  attr)
inline

◆ setPropertiesFromAttr()

llvm::LogicalResult mlir::tt::ttnn::ArangeOp::setPropertiesFromAttr ( Properties prop,
::mlir::Attribute  attr,
::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError 
)
static

◆ setStart()

void mlir::tt::ttnn::ArangeOp::setStart ( uint64_t  attrValue)

◆ setStartAttr()

void mlir::tt::ttnn::ArangeOp::setStartAttr ( ::mlir::IntegerAttr  attr)
inline

◆ setStep()

void mlir::tt::ttnn::ArangeOp::setStep ( uint64_t  attrValue)

◆ setStepAttr()

void mlir::tt::ttnn::ArangeOp::setStepAttr ( ::mlir::IntegerAttr  attr)
inline

◆ verify()

::llvm::LogicalResult mlir::tt::ttnn::ArangeOp::verify ( )

◆ verifyInherentAttrs()

llvm::LogicalResult mlir::tt::ttnn::ArangeOp::verifyInherentAttrs ( ::mlir::OperationName  opName,
::mlir::NamedAttrList &  attrs,
llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError 
)
static

◆ verifyInvariants()

llvm::LogicalResult mlir::tt::ttnn::ArangeOp::verifyInvariants ( )

◆ verifyInvariantsImpl()

llvm::LogicalResult mlir::tt::ttnn::ArangeOp::verifyInvariantsImpl ( )

◆ writeProperties()

void mlir::tt::ttnn::ArangeOp::writeProperties ( ::mlir::DialectBytecodeWriter &  writer)

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