mlir::tt::ttnn::EmbeddingOp Class Reference

Inheritance diagram for mlir::tt::ttnn::EmbeddingOp:

Public Types
using	Adaptor = EmbeddingOpAdaptor
template<typename RangeT >
using	GenericAdaptor = EmbeddingOpGenericAdaptor< 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 >	getInput ()
::mlir::TypedValue<::mlir::RankedTensorType >	getWeight ()
::mlir::TypedValue<::mlir::RankedTensorType >	getOutput ()
::mlir::OpOperand &	getInputMutable ()
::mlir::OpOperand &	getWeightMutable ()
::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 input, ::mlir::Value weight, ::mlir::Value output)
static void	build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::Value weight, ::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

using mlir::tt::ttnn::EmbeddingOp::Adaptor = EmbeddingOpAdaptor

FoldAdaptor

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

GenericAdaptor

template<typename RangeT >

using mlir::tt::ttnn::EmbeddingOp::GenericAdaptor = EmbeddingOpGenericAdaptor<RangeT>

Member Function Documentation

build() [1/3]

void mlir::tt::ttnn::EmbeddingOp::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::EmbeddingOp::build ( ::mlir::OpBuilder &  odsBuilder, ::mlir::OperationState &  odsState, ::mlir::Type  result, ::mlir::Value  input, ::mlir::Value  weight, ::mlir::Value  output  )

static

build() [3/3]

void mlir::tt::ttnn::EmbeddingOp::build ( ::mlir::OpBuilder &  odsBuilder, ::mlir::OperationState &  odsState, ::mlir::TypeRange  resultTypes, ::mlir::Value  input, ::mlir::Value  weight, ::mlir::Value  output  )

static

getAttributeNames()

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

inline

getDpsInitsMutable()

MutableOperandRange mlir::tt::ttnn::EmbeddingOp::getDpsInitsMutable ( )

inline

getInput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::EmbeddingOp::getInput ( )

inline

getInputMutable()

::mlir::OpOperand& mlir::tt::ttnn::EmbeddingOp::getInputMutable ( )

inline

getODSOperandIndexAndLength()

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

inline

getODSOperands()

::mlir::Operation::operand_range mlir::tt::ttnn::EmbeddingOp::getODSOperands ( unsigned  index )

inline

getODSResultIndexAndLength()

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

inline

getODSResults()

::mlir::Operation::result_range mlir::tt::ttnn::EmbeddingOp::getODSResults ( unsigned  index )

inline

getOperationName()

static constexpr ::llvm::StringLiteral mlir::tt::ttnn::EmbeddingOp::getOperationName ( )

inlinestatic

getOutput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::EmbeddingOp::getOutput ( )

inline

getOutputMutable()

::mlir::OpOperand& mlir::tt::ttnn::EmbeddingOp::getOutputMutable ( )

inline

getResult()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::EmbeddingOp::getResult ( )

inline

getWeight()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttnn::EmbeddingOp::getWeight ( )

inline

getWeightMutable()

::mlir::OpOperand& mlir::tt::ttnn::EmbeddingOp::getWeightMutable ( )

inline

verify()

::llvm::LogicalResult mlir::tt::ttnn::EmbeddingOp::verify

(

)

verifyInvariants()

llvm::LogicalResult mlir::tt::ttnn::EmbeddingOp::verifyInvariants

(

)

verifyInvariantsImpl()

llvm::LogicalResult mlir::tt::ttnn::EmbeddingOp::verifyInvariantsImpl

(

)

---

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

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTNN/IR/TTNNOps.h.inc
• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTNN/IR/TTNNOps.cpp.inc