TT-MLIR
|
Public Types | |
using | Adaptor = ReduceTileOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = ReduceTileOpGenericAdaptor< RangeT > |
using | FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > > |
using | Properties = FoldAdaptor::Properties |
Public Member Functions | |
static ::llvm::ArrayRef<::llvm::StringRef > | getAttributeNames () |
::mlir::StringAttr | getReduceDimAttrName () |
::mlir::StringAttr | getReduceTypeAttrName () |
std::pair< unsigned, unsigned > | getODSOperandIndexAndLength (unsigned index) |
::mlir::Operation::operand_range | getODSOperands (unsigned index) |
::mlir::TypedValue<::mlir::tt::ttkernel::CBType > | getInCb () |
::mlir::TypedValue<::mlir::tt::ttkernel::CBType > | getScalingCb () |
::mlir::TypedValue<::mlir::IntegerType > | getInTileIndex () |
::mlir::TypedValue<::mlir::IntegerType > | getScalingTileIndex () |
::mlir::TypedValue<::mlir::IntegerType > | getDstIndex () |
::mlir::OpOperand & | getInCbMutable () |
::mlir::OpOperand & | getScalingCbMutable () |
::mlir::OpOperand & | getInTileIndexMutable () |
::mlir::OpOperand & | getScalingTileIndexMutable () |
::mlir::OpOperand & | getDstIndexMutable () |
std::pair< unsigned, unsigned > | getODSResultIndexAndLength (unsigned index) |
::mlir::Operation::result_range | getODSResults (unsigned index) |
void | writeProperties (::mlir::DialectBytecodeWriter &writer) |
::mlir::tt::ttkernel::ReduceTypeAttr | getReduceTypeAttr () |
::mlir::tt::ttkernel::ReduceType | getReduceType () |
::mlir::tt::ttkernel::ReduceDimAttr | getReduceDimAttr () |
::mlir::tt::ttkernel::ReduceDim | getReduceDim () |
void | setReduceTypeAttr (::mlir::tt::ttkernel::ReduceTypeAttr attr) |
void | setReduceType (::mlir::tt::ttkernel::ReduceType attrValue) |
void | setReduceDimAttr (::mlir::tt::ttkernel::ReduceDimAttr attr) |
void | setReduceDim (::mlir::tt::ttkernel::ReduceDim attrValue) |
::llvm::LogicalResult | verifyInvariantsImpl () |
::llvm::LogicalResult | verifyInvariants () |
Static Public Member Functions | |
::mlir::StringAttr | getReduceDimAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getReduceTypeAttrName (::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::Value in_cb, ::mlir::Value scaling_cb, ::mlir::Value in_tile_index, ::mlir::Value scaling_tile_index, ::mlir::Value dst_index, ::mlir::tt::ttkernel::ReduceTypeAttr reduce_type, ::mlir::tt::ttkernel::ReduceDimAttr reduce_dim) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value in_cb, ::mlir::Value scaling_cb, ::mlir::Value in_tile_index, ::mlir::Value scaling_tile_index, ::mlir::Value dst_index, ::mlir::tt::ttkernel::ReduceTypeAttr reduce_type, ::mlir::tt::ttkernel::ReduceDimAttr reduce_dim) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value in_cb, ::mlir::Value scaling_cb, ::mlir::Value in_tile_index, ::mlir::Value scaling_tile_index, ::mlir::Value dst_index, ::mlir::tt::ttkernel::ReduceType reduce_type, ::mlir::tt::ttkernel::ReduceDim reduce_dim) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value in_cb, ::mlir::Value scaling_cb, ::mlir::Value in_tile_index, ::mlir::Value scaling_tile_index, ::mlir::Value dst_index, ::mlir::tt::ttkernel::ReduceType reduce_type, ::mlir::tt::ttkernel::ReduceDim reduce_dim) |
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::ttkernel::ReduceTileOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttkernel::ReduceTileOp::GenericAdaptor = ReduceTileOpGenericAdaptor<RangeT> |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
static |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inlinestatic |
|
static |
mlir::tt::ttkernel::ReduceDim mlir::tt::ttkernel::ReduceTileOp::getReduceDim | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
mlir::tt::ttkernel::ReduceType mlir::tt::ttkernel::ReduceTileOp::getReduceType | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
inline |
|
inline |
|
static |
|
static |
|
static |
|
static |
void mlir::tt::ttkernel::ReduceTileOp::setReduceDim | ( | ::mlir::tt::ttkernel::ReduceDim | attrValue | ) |
|
inline |
void mlir::tt::ttkernel::ReduceTileOp::setReduceType | ( | ::mlir::tt::ttkernel::ReduceType | attrValue | ) |
|
inline |
|
static |
llvm::LogicalResult mlir::tt::ttkernel::ReduceTileOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttkernel::ReduceTileOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttkernel::ReduceTileOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |