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

Public Types

using Adaptor = LinearOpAdaptor
 
template<typename RangeT >
using GenericAdaptor = LinearOpGenericAdaptor< RangeT >
 
using FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > >
 

Public Member Functions

static ::llvm::ArrayRef<::llvm::StringRef > getAttributeNames ()
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index)
 
::mlir::Operation::operand_range getODSOperands (unsigned index)
 
::mlir::TypedValue<::mlir::RankedTensorType > getA ()
 
::mlir::TypedValue<::mlir::RankedTensorType > getB ()
 
::mlir::TypedValue<::mlir::RankedTensorType > getBias ()
 
::mlir::TypedValue<::mlir::RankedTensorType > getOutput ()
 
::mlir::OpOperand & getAMutable ()
 
::mlir::OpOperand & getBMutable ()
 
::mlir::MutableOperandRange getBiasMutable ()
 
::mlir::OpOperand & getOutputMutable ()
 
std::pair< unsigned, unsigned > getODSResultIndexAndLength (unsigned index)
 
::mlir::Operation::result_range getODSResults (unsigned index)
 
::mlir::TypedValue<::mlir::RankedTensorType > getResult ()
 
::llvm::LogicalResult verifyInvariantsImpl ()
 
::llvm::LogicalResult verifyInvariants ()
 
::llvm::LogicalResult verify ()
 
MutableOperandRange getDpsInitsMutable ()
 

Static Public Member Functions

static constexpr ::llvm::StringLiteral getOperationName ()
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value a, ::mlir::Value b, ::mlir::Value bias, ::mlir::Value output)
 
static void build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value a, ::mlir::Value b, ::mlir::Value bias, ::mlir::Value output)
 
static void build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute > attributes={})
 

Member Typedef Documentation

◆ Adaptor

◆ FoldAdaptor

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

◆ GenericAdaptor

template<typename RangeT >
using mlir::tt::ttnn::LinearOp::GenericAdaptor = LinearOpGenericAdaptor<RangeT>

Member Function Documentation

◆ build() [1/3]

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

◆ build() [2/3]

void mlir::tt::ttnn::LinearOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::Type  result,
::mlir::Value  a,
::mlir::Value  b,
::mlir::Value  bias,
::mlir::Value  output 
)
static

◆ build() [3/3]

void mlir::tt::ttnn::LinearOp::build ( ::mlir::OpBuilder &  odsBuilder,
::mlir::OperationState &  odsState,
::mlir::TypeRange  resultTypes,
::mlir::Value  a,
::mlir::Value  b,
::mlir::Value  bias,
::mlir::Value  output 
)
static

◆ getA()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::LinearOp::getA ( )
inline

◆ getAMutable()

::mlir::OpOperand& mlir::tt::ttnn::LinearOp::getAMutable ( )
inline

◆ getAttributeNames()

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

◆ getB()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::LinearOp::getB ( )
inline

◆ getBias()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::LinearOp::getBias ( )
inline

◆ getBiasMutable()

mlir::MutableOperandRange mlir::tt::ttnn::LinearOp::getBiasMutable ( )

◆ getBMutable()

::mlir::OpOperand& mlir::tt::ttnn::LinearOp::getBMutable ( )
inline

◆ getDpsInitsMutable()

MutableOperandRange mlir::tt::ttnn::LinearOp::getDpsInitsMutable ( )
inline

◆ getODSOperandIndexAndLength()

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

◆ getODSOperands()

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

◆ getODSResultIndexAndLength()

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

◆ getODSResults()

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

◆ getOperationName()

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

◆ getOutput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::LinearOp::getOutput ( )
inline

◆ getOutputMutable()

::mlir::OpOperand& mlir::tt::ttnn::LinearOp::getOutputMutable ( )
inline

◆ getResult()

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

◆ verify()

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

◆ verifyInvariants()

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

◆ verifyInvariantsImpl()

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

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