TT-MLIR
|
Public Types | |
using | Adaptor = ScatterOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = ScatterOpGenericAdaptor< RangeT > |
using | FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > > |
using | Properties = FoldAdaptor::Properties |
Public Member Functions | |
static ::llvm::ArrayRef<::llvm::StringRef > | getAttributeNames () |
::mlir::StringAttr | getIndexVectorDimAttrName () |
::mlir::StringAttr | getIndicesAreSortedAttrName () |
::mlir::StringAttr | getInputBatchingDimsAttrName () |
::mlir::StringAttr | getInsertedWindowDimsAttrName () |
::mlir::StringAttr | getScatterDimsToOperandDimsAttrName () |
::mlir::StringAttr | getScatterIndicesBatchingDimsAttrName () |
::mlir::StringAttr | getUniqueIndicesAttrName () |
::mlir::StringAttr | getUpdateWindowDimsAttrName () |
std::pair< unsigned, unsigned > | getODSOperandIndexAndLength (unsigned index) |
::mlir::Operation::operand_range | getODSOperands (unsigned index) |
::mlir::TypedValue<::mlir::RankedTensorType > | getInput () |
::mlir::TypedValue<::mlir::RankedTensorType > | getScatterIndices () |
::mlir::TypedValue<::mlir::RankedTensorType > | getUpdate () |
::mlir::TypedValue<::mlir::RankedTensorType > | getOutput () |
::mlir::OpOperand & | getInputMutable () |
::mlir::OpOperand & | getScatterIndicesMutable () |
::mlir::OpOperand & | getUpdateMutable () |
::mlir::OpOperand & | getOutputMutable () |
std::pair< unsigned, unsigned > | getODSResultIndexAndLength (unsigned index) |
::mlir::Operation::result_range | getODSResults (unsigned index) |
::mlir::TypedValue<::mlir::RankedTensorType > | getResult () |
::mlir::Region & | getUpdateComputation () |
void | writeProperties (::mlir::DialectBytecodeWriter &writer) |
::mlir::DenseI32ArrayAttr | getUpdateWindowDimsAttr () |
::llvm::ArrayRef< int32_t > | getUpdateWindowDims () |
::mlir::DenseI32ArrayAttr | getInsertedWindowDimsAttr () |
::llvm::ArrayRef< int32_t > | getInsertedWindowDims () |
::mlir::DenseI32ArrayAttr | getInputBatchingDimsAttr () |
::llvm::ArrayRef< int32_t > | getInputBatchingDims () |
::mlir::DenseI32ArrayAttr | getScatterIndicesBatchingDimsAttr () |
::llvm::ArrayRef< int32_t > | getScatterIndicesBatchingDims () |
::mlir::DenseI32ArrayAttr | getScatterDimsToOperandDimsAttr () |
::llvm::ArrayRef< int32_t > | getScatterDimsToOperandDims () |
::mlir::IntegerAttr | getIndexVectorDimAttr () |
uint32_t | getIndexVectorDim () |
::mlir::BoolAttr | getIndicesAreSortedAttr () |
bool | getIndicesAreSorted () |
::mlir::BoolAttr | getUniqueIndicesAttr () |
bool | getUniqueIndices () |
void | setUpdateWindowDimsAttr (::mlir::DenseI32ArrayAttr attr) |
void | setUpdateWindowDims (::llvm::ArrayRef< int32_t > attrValue) |
void | setInsertedWindowDimsAttr (::mlir::DenseI32ArrayAttr attr) |
void | setInsertedWindowDims (::llvm::ArrayRef< int32_t > attrValue) |
void | setInputBatchingDimsAttr (::mlir::DenseI32ArrayAttr attr) |
void | setInputBatchingDims (::llvm::ArrayRef< int32_t > attrValue) |
void | setScatterIndicesBatchingDimsAttr (::mlir::DenseI32ArrayAttr attr) |
void | setScatterIndicesBatchingDims (::llvm::ArrayRef< int32_t > attrValue) |
void | setScatterDimsToOperandDimsAttr (::mlir::DenseI32ArrayAttr attr) |
void | setScatterDimsToOperandDims (::llvm::ArrayRef< int32_t > attrValue) |
void | setIndexVectorDimAttr (::mlir::IntegerAttr attr) |
void | setIndexVectorDim (uint32_t attrValue) |
void | setIndicesAreSortedAttr (::mlir::BoolAttr attr) |
void | setIndicesAreSorted (bool attrValue) |
void | setUniqueIndicesAttr (::mlir::BoolAttr attr) |
void | setUniqueIndices (bool attrValue) |
::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 | |
::mlir::StringAttr | getIndexVectorDimAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getIndicesAreSortedAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputBatchingDimsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInsertedWindowDimsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getScatterDimsToOperandDimsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getScatterIndicesBatchingDimsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getUniqueIndicesAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getUpdateWindowDimsAttrName (::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::Value scatter_indices, ::mlir::Value update, ::mlir::DenseI32ArrayAttr update_window_dims, ::mlir::DenseI32ArrayAttr inserted_window_dims, ::mlir::DenseI32ArrayAttr input_batching_dims, ::mlir::DenseI32ArrayAttr scatter_indices_batching_dims, ::mlir::DenseI32ArrayAttr scatter_dims_to_operand_dims, ::mlir::IntegerAttr index_vector_dim, ::mlir::BoolAttr indices_are_sorted, ::mlir::BoolAttr unique_indices, ::mlir::Value output) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::Value scatter_indices, ::mlir::Value update, ::mlir::DenseI32ArrayAttr update_window_dims, ::mlir::DenseI32ArrayAttr inserted_window_dims, ::mlir::DenseI32ArrayAttr input_batching_dims, ::mlir::DenseI32ArrayAttr scatter_indices_batching_dims, ::mlir::DenseI32ArrayAttr scatter_dims_to_operand_dims, ::mlir::IntegerAttr index_vector_dim, ::mlir::BoolAttr indices_are_sorted, ::mlir::BoolAttr unique_indices, ::mlir::Value output) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value input, ::mlir::Value scatter_indices, ::mlir::Value update, ::llvm::ArrayRef< int32_t > update_window_dims, ::llvm::ArrayRef< int32_t > inserted_window_dims, ::llvm::ArrayRef< int32_t > input_batching_dims, ::llvm::ArrayRef< int32_t > scatter_indices_batching_dims, ::llvm::ArrayRef< int32_t > scatter_dims_to_operand_dims, uint32_t index_vector_dim, bool indices_are_sorted, bool unique_indices, ::mlir::Value output) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::Value scatter_indices, ::mlir::Value update, ::llvm::ArrayRef< int32_t > update_window_dims, ::llvm::ArrayRef< int32_t > inserted_window_dims, ::llvm::ArrayRef< int32_t > input_batching_dims, ::llvm::ArrayRef< int32_t > scatter_indices_batching_dims, ::llvm::ArrayRef< int32_t > scatter_dims_to_operand_dims, uint32_t index_vector_dim, bool indices_are_sorted, bool unique_indices, ::mlir::Value output) |
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) |
using mlir::tt::ttir::ScatterOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttir::ScatterOp::GenericAdaptor = ScatterOpGenericAdaptor<RangeT> |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
|
inline |
void mlir::tt::ttir::ScatterOp::getEffects | ( | ::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> & | effects | ) |
uint32_t mlir::tt::ttir::ScatterOp::getIndexVectorDim | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
bool mlir::tt::ttir::ScatterOp::getIndicesAreSorted | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
llvm::ArrayRef< int32_t > mlir::tt::ttir::ScatterOp::getInputBatchingDims | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
llvm::ArrayRef< int32_t > mlir::tt::ttir::ScatterOp::getInsertedWindowDims | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
static |
|
inline |
llvm::ArrayRef< int32_t > mlir::tt::ttir::ScatterOp::getScatterDimsToOperandDims | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
llvm::ArrayRef< int32_t > mlir::tt::ttir::ScatterOp::getScatterIndicesBatchingDims | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
bool mlir::tt::ttir::ScatterOp::getUniqueIndices | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
inline |
llvm::ArrayRef< int32_t > mlir::tt::ttir::ScatterOp::getUpdateWindowDims | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
static |
void mlir::tt::ttir::ScatterOp::setIndexVectorDim | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttir::ScatterOp::setIndicesAreSorted | ( | bool | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttir::ScatterOp::setInputBatchingDims | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
void mlir::tt::ttir::ScatterOp::setInsertedWindowDims | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttir::ScatterOp::setScatterDimsToOperandDims | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
void mlir::tt::ttir::ScatterOp::setScatterIndicesBatchingDims | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
void mlir::tt::ttir::ScatterOp::setUniqueIndices | ( | bool | attrValue | ) |
|
inline |
void mlir::tt::ttir::ScatterOp::setUpdateWindowDims | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
::llvm::LogicalResult mlir::tt::ttir::ScatterOp::verify | ( | ) |
|
static |
llvm::LogicalResult mlir::tt::ttir::ScatterOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttir::ScatterOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttir::ScatterOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |