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::MeanOp Class Reference
Inheritance diagram for mlir::tt::ttnn::MeanOp:

Public Types

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

Public Member Functions

static ::llvm::ArrayRef<::llvm::StringRef > getAttributeNames ()
 
::mlir::StringAttr getDimArgAttrName ()
 
::mlir::StringAttr getKeepDimAttrName ()
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index)
 
::mlir::Operation::operand_range getODSOperands (unsigned index)
 
::mlir::TypedValue<::mlir::RankedTensorType > getInput ()
 
::mlir::OpOperand & getInputMutable ()
 
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::BoolAttr getKeepDimAttr ()
 
bool getKeepDim ()
 
::mlir::ArrayAttr getDimArgAttr ()
 
::std::optional< ::mlir::ArrayAttr > getDimArg ()
 
void setKeepDimAttr (::mlir::BoolAttr attr)
 
void setKeepDim (bool attrValue)
 
void setDimArgAttr (::mlir::ArrayAttr attr)
 
::mlir::Attribute removeDimArgAttr ()
 
::llvm::LogicalResult verifyInvariantsImpl ()
 
::llvm::LogicalResult verifyInvariants ()
 

Static Public Member Functions

::mlir::StringAttr getDimArgAttrName (::mlir::OperationName name)
 
::mlir::StringAttr getKeepDimAttrName (::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::Value input, ::mlir::BoolAttr keep_dim, ::mlir::ArrayAttr dim_arg)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::BoolAttr keep_dim, ::mlir::ArrayAttr dim_arg)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value input, bool keep_dim, ::mlir::ArrayAttr dim_arg)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, bool keep_dim, ::mlir::ArrayAttr dim_arg)
 
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::MeanOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> >

◆ GenericAdaptor

template<typename RangeT >
using mlir::tt::ttnn::MeanOp::GenericAdaptor = MeanOpGenericAdaptor<RangeT>

◆ Properties

Member Function Documentation

◆ build() [1/5]

void mlir::tt::ttnn::MeanOp::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::MeanOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::Type  result,
::mlir::Value  input,
::mlir::BoolAttr  keep_dim,
::mlir::ArrayAttr  dim_arg 
)
static

◆ build() [3/5]

void mlir::tt::ttnn::MeanOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::Type  result,
::mlir::Value  input,
bool  keep_dim,
::mlir::ArrayAttr  dim_arg 
)
static

◆ build() [4/5]

void mlir::tt::ttnn::MeanOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::TypeRange  resultTypes,
::mlir::Value  input,
::mlir::BoolAttr  keep_dim,
::mlir::ArrayAttr  dim_arg 
)
static

◆ build() [5/5]

void mlir::tt::ttnn::MeanOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::TypeRange  resultTypes,
::mlir::Value  input,
bool  keep_dim,
::mlir::ArrayAttr  dim_arg 
)
static

◆ computePropertiesHash()

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

◆ getAttributeNameForIndex() [1/2]

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

◆ getAttributeNameForIndex() [2/2]

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

◆ getAttributeNames()

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

◆ getDimArg()

std::optional<::mlir::ArrayAttr > mlir::tt::ttnn::MeanOp::getDimArg ( )

◆ getDimArgAttr()

::mlir::ArrayAttr mlir::tt::ttnn::MeanOp::getDimArgAttr ( )
inline

◆ getDimArgAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttnn::MeanOp::getDimArgAttrName ( )
inline

◆ getDimArgAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttnn::MeanOp::getDimArgAttrName ( ::mlir::OperationName  name)
inlinestatic

◆ getInherentAttr()

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

◆ getInput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::MeanOp::getInput ( )
inline

◆ getInputMutable()

::mlir::OpOperand& mlir::tt::ttnn::MeanOp::getInputMutable ( )
inline

◆ getKeepDim()

bool mlir::tt::ttnn::MeanOp::getKeepDim ( )

◆ getKeepDimAttr()

::mlir::BoolAttr mlir::tt::ttnn::MeanOp::getKeepDimAttr ( )
inline

◆ getKeepDimAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttnn::MeanOp::getKeepDimAttrName ( )
inline

◆ getKeepDimAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttnn::MeanOp::getKeepDimAttrName ( ::mlir::OperationName  name)
inlinestatic

◆ getODSOperandIndexAndLength()

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

◆ getODSOperands()

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

◆ getODSResultIndexAndLength()

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

◆ getODSResults()

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

◆ getOperationName()

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

◆ getPropertiesAsAttr()

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

◆ getResult()

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

◆ populateInherentAttrs()

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

◆ readProperties()

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

◆ removeDimArgAttr()

::mlir::Attribute mlir::tt::ttnn::MeanOp::removeDimArgAttr ( )
inline

◆ setDimArgAttr()

void mlir::tt::ttnn::MeanOp::setDimArgAttr ( ::mlir::ArrayAttr  attr)
inline

◆ setInherentAttr()

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

◆ setKeepDim()

void mlir::tt::ttnn::MeanOp::setKeepDim ( bool  attrValue)

◆ setKeepDimAttr()

void mlir::tt::ttnn::MeanOp::setKeepDimAttr ( ::mlir::BoolAttr  attr)
inline

◆ setPropertiesFromAttr()

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

◆ verifyInherentAttrs()

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

◆ verifyInvariants()

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

◆ verifyInvariantsImpl()

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

◆ writeProperties()

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

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