mlir::tt::ttir::EmbeddingBackwardOp Class Reference

Inheritance diagram for mlir::tt::ttir::EmbeddingBackwardOp:

Public Types
using	Adaptor = EmbeddingBackwardOpAdaptor
template<typename RangeT >
using	GenericAdaptor = EmbeddingBackwardOpGenericAdaptor< 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 >	getInGradient ()
::mlir::TypedValue<::mlir::RankedTensorType >	getOutput ()
::mlir::OpOperand &	getInputMutable ()
::mlir::OpOperand &	getWeightMutable ()
::mlir::OpOperand &	getInGradientMutable ()
::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 ()
void	getEffects (::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> &effects)
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 in_gradient, ::mlir::Value output)
static void	build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::Value weight, ::mlir::Value in_gradient, ::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::ttir::EmbeddingBackwardOp::Adaptor = EmbeddingBackwardOpAdaptor

FoldAdaptor

using mlir::tt::ttir::EmbeddingBackwardOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> >

GenericAdaptor

template<typename RangeT >

using mlir::tt::ttir::EmbeddingBackwardOp::GenericAdaptor = EmbeddingBackwardOpGenericAdaptor<RangeT>

Member Function Documentation

build() [1/3]

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

static

build() [2/3]

void mlir::tt::ttir::EmbeddingBackwardOp::build ( ::mlir::OpBuilder &  odsBuilder, ::mlir::OperationState &  odsState, ::mlir::Type  result, ::mlir::Value  input, ::mlir::Value  weight, ::mlir::Value  in_gradient, ::mlir::Value  output  )

static

build() [3/3]

void mlir::tt::ttir::EmbeddingBackwardOp::build ( ::mlir::OpBuilder &  odsBuilder, ::mlir::OperationState &  odsState, ::mlir::TypeRange  resultTypes, ::mlir::Value  input, ::mlir::Value  weight, ::mlir::Value  in_gradient, ::mlir::Value  output  )

static

getAttributeNames()

static ::llvm::ArrayRef<::llvm::StringRef> mlir::tt::ttir::EmbeddingBackwardOp::getAttributeNames ( )

inline

getDpsInitsMutable()

MutableOperandRange mlir::tt::ttir::EmbeddingBackwardOp::getDpsInitsMutable ( )

inline

getEffects()

void mlir::tt::ttir::EmbeddingBackwardOp::getEffects

(

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

effects

)

getInGradient()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::EmbeddingBackwardOp::getInGradient ( )

inline

getInGradientMutable()

::mlir::OpOperand& mlir::tt::ttir::EmbeddingBackwardOp::getInGradientMutable ( )

inline

getInput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::EmbeddingBackwardOp::getInput ( )

inline

getInputMutable()

::mlir::OpOperand& mlir::tt::ttir::EmbeddingBackwardOp::getInputMutable ( )

inline

getODSOperandIndexAndLength()

std::pair<unsigned, unsigned> mlir::tt::ttir::EmbeddingBackwardOp::getODSOperandIndexAndLength ( unsigned  index )

inline

getODSOperands()

::mlir::Operation::operand_range mlir::tt::ttir::EmbeddingBackwardOp::getODSOperands ( unsigned  index )

inline

getODSResultIndexAndLength()

std::pair<unsigned, unsigned> mlir::tt::ttir::EmbeddingBackwardOp::getODSResultIndexAndLength ( unsigned  index )

inline

getODSResults()

::mlir::Operation::result_range mlir::tt::ttir::EmbeddingBackwardOp::getODSResults ( unsigned  index )

inline

getOperationName()

static constexpr ::llvm::StringLiteral mlir::tt::ttir::EmbeddingBackwardOp::getOperationName ( )

inlinestatic

getOutput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::EmbeddingBackwardOp::getOutput ( )

inline

getOutputMutable()

::mlir::OpOperand& mlir::tt::ttir::EmbeddingBackwardOp::getOutputMutable ( )

inline

getResult()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::EmbeddingBackwardOp::getResult ( )

inline

getWeight()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::EmbeddingBackwardOp::getWeight ( )

inline

getWeightMutable()

::mlir::OpOperand& mlir::tt::ttir::EmbeddingBackwardOp::getWeightMutable ( )

inline

verify()

::llvm::LogicalResult mlir::tt::ttir::EmbeddingBackwardOp::verify

(

)

verifyInvariants()

llvm::LogicalResult mlir::tt::ttir::EmbeddingBackwardOp::verifyInvariants

(

)

verifyInvariantsImpl()

llvm::LogicalResult mlir::tt::ttir::EmbeddingBackwardOp::verifyInvariantsImpl

(

)

---

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

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTIR/IR/TTIROps.h.inc
• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTIR/IR/TTIROps.cpp.inc