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 > >
using	Properties = FoldAdaptor::Properties

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::OpOperand &	getInputMutable ()
::mlir::OpOperand &	getWeightMutable ()
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 ()
llvm::Expected< size_t >	getOpRuntime (const std::vector< TTNNLayoutAttr > &inputs, const OpConfig &config)
llvm::Expected< mlir::tt::op_model::ttnn::OpConstraints >	getOpConstraints (const std::vector< TTNNLayoutAttr > &inputs, const OpConfig &config)
void	getEffects (::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> &effects)
wa::TTNNOperandsWorkarounds	getOperandsWorkarounds ()

Static Public Member Functions
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)
static void	build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value input, ::mlir::Value weight)
static void	build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::Value weight)
static void	build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute > attributes={})
static void	build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, const Properties &properties, ::llvm::ArrayRef<::mlir::NamedAttribute > discardableAttributes={})

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>

Properties

using mlir::tt::ttnn::EmbeddingOp::Properties = FoldAdaptor::Properties

Member Function Documentation

build() [1/4]

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/4]

void mlir::tt::ttnn::EmbeddingOp::build ( ::mlir::OpBuilder &  , ::mlir::OperationState &  odsState, ::mlir::TypeRange  resultTypes, ::mlir::ValueRange  operands, const Properties &  properties, ::llvm::ArrayRef<::mlir::NamedAttribute >  discardableAttributes = {}  )

static

build() [3/4]

void mlir::tt::ttnn::EmbeddingOp::build ( ::mlir::OpBuilder &  odsBuilder, ::mlir::OperationState &  odsState, ::mlir::Type  result, ::mlir::Value  input, ::mlir::Value  weight  )

static

build() [4/4]

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

static

computePropertiesHash()

llvm::hash_code mlir::tt::ttnn::EmbeddingOp::computePropertiesHash ( const Properties &  prop )

static

getAttributeNames()

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

inline

getEffects()

void mlir::tt::ttnn::EmbeddingOp::getEffects

(

::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> &

effects

)

getInherentAttr()

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

static

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

getOpConstraints()

llvm::Expected<mlir::tt::op_model::ttnn::OpConstraints> mlir::tt::ttnn::EmbeddingOp::getOpConstraints	(	const std::vector< TTNNLayoutAttr > &	inputs,
		const OpConfig &	config
	)

getOperandsWorkarounds()

wa::TTNNOperandsWorkarounds mlir::tt::ttnn::EmbeddingOp::getOperandsWorkarounds ( )

inline

getOperationName()

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

inlinestatic

getOpRuntime()

llvm::Expected<size_t> mlir::tt::ttnn::EmbeddingOp::getOpRuntime	(	const std::vector< TTNNLayoutAttr > &	inputs,
		const OpConfig &	config
	)

getPropertiesAsAttr()

mlir::Attribute mlir::tt::ttnn::EmbeddingOp::getPropertiesAsAttr ( ::mlir::MLIRContext *  ctx, const Properties &  prop  )

static

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

populateInherentAttrs()

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

static

setInherentAttr()

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

static

setPropertiesFromAttr()

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

static

verify()

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

(

)

verifyInherentAttrs()

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

static

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