TT-MLIR
|
Public Types | |
using | Adaptor = ConvolutionOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = ConvolutionOpGenericAdaptor< RangeT > |
using | FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > > |
using | Properties = FoldAdaptor::Properties |
Public Member Functions | |
static ::llvm::ArrayRef<::llvm::StringRef > | getAttributeNames () |
::mlir::StringAttr | getBatchGroupCountAttrName () |
::mlir::StringAttr | getConvolutionLayoutAttrName () |
::mlir::StringAttr | getFeatureGroupCountAttrName () |
::mlir::StringAttr | getInputDilationAttrName () |
::mlir::StringAttr | getOperandConstraintsAttrName () |
::mlir::StringAttr | getPaddingAttrName () |
::mlir::StringAttr | getWeightDilationAttrName () |
::mlir::StringAttr | getWindowReversalAttrName () |
::mlir::StringAttr | getWindowStridesAttrName () |
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 > | getBias () |
::mlir::TypedValue<::mlir::RankedTensorType > | getOutput () |
::mlir::OpOperand & | getInputMutable () |
::mlir::OpOperand & | getWeightMutable () |
::mlir::MutableOperandRange | getBiasMutable () |
::mlir::OpOperand & | getOutputMutable () |
std::pair< unsigned, unsigned > | getODSResultIndexAndLength (unsigned index) |
::mlir::Operation::result_range | getODSResults (unsigned index) |
void | writeProperties (::mlir::DialectBytecodeWriter &writer) |
::mlir::DenseI64ArrayAttr | getWindowStridesAttr () |
::llvm::ArrayRef< int64_t > | getWindowStrides () |
::mlir::DenseI64ArrayAttr | getPaddingAttr () |
::llvm::ArrayRef< int64_t > | getPadding () |
::mlir::DenseI64ArrayAttr | getInputDilationAttr () |
::llvm::ArrayRef< int64_t > | getInputDilation () |
::mlir::DenseI64ArrayAttr | getWeightDilationAttr () |
::llvm::ArrayRef< int64_t > | getWeightDilation () |
::mlir::DenseBoolArrayAttr | getWindowReversalAttr () |
::llvm::ArrayRef< bool > | getWindowReversal () |
::mlir::tt::ttir::ConvolutionLayoutAttr | getConvolutionLayoutAttr () |
::mlir::tt::ttir::ConvolutionLayoutAttr | getConvolutionLayout () |
::mlir::IntegerAttr | getFeatureGroupCountAttr () |
uint64_t | getFeatureGroupCount () |
::mlir::IntegerAttr | getBatchGroupCountAttr () |
uint64_t | getBatchGroupCount () |
::mlir::ArrayAttr | getOperandConstraintsAttr () |
::mlir::ArrayAttr | getOperandConstraints () |
void | setWindowStridesAttr (::mlir::DenseI64ArrayAttr attr) |
void | setWindowStrides (::llvm::ArrayRef< int64_t > attrValue) |
void | setPaddingAttr (::mlir::DenseI64ArrayAttr attr) |
void | setPadding (::llvm::ArrayRef< int64_t > attrValue) |
void | setInputDilationAttr (::mlir::DenseI64ArrayAttr attr) |
void | setInputDilation (::llvm::ArrayRef< int64_t > attrValue) |
void | setWeightDilationAttr (::mlir::DenseI64ArrayAttr attr) |
void | setWeightDilation (::llvm::ArrayRef< int64_t > attrValue) |
void | setWindowReversalAttr (::mlir::DenseBoolArrayAttr attr) |
void | setWindowReversal (::llvm::ArrayRef< bool > attrValue) |
void | setConvolutionLayoutAttr (::mlir::tt::ttir::ConvolutionLayoutAttr attr) |
void | setFeatureGroupCountAttr (::mlir::IntegerAttr attr) |
void | setFeatureGroupCount (uint64_t attrValue) |
void | setBatchGroupCountAttr (::mlir::IntegerAttr attr) |
void | setBatchGroupCount (uint64_t attrValue) |
void | setOperandConstraintsAttr (::mlir::ArrayAttr attr) |
::llvm::LogicalResult | verifyInvariantsImpl () |
::llvm::LogicalResult | verifyInvariants () |
::llvm::LogicalResult | verify () |
MutableOperandRange | getDpsInitsMutable () |
Static Public Member Functions | |
::mlir::StringAttr | getBatchGroupCountAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getConvolutionLayoutAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getFeatureGroupCountAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputDilationAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getOperandConstraintsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getPaddingAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getWeightDilationAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getWindowReversalAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getWindowStridesAttrName (::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 resultType0, ::mlir::Value input, ::mlir::Value weight, ::mlir::Value bias, ::mlir::Value output, ::mlir::DenseI64ArrayAttr window_strides, ::mlir::DenseI64ArrayAttr padding, ::mlir::DenseI64ArrayAttr input_dilation, ::mlir::DenseI64ArrayAttr weight_dilation, ::mlir::DenseBoolArrayAttr window_reversal, ::mlir::tt::ttir::ConvolutionLayoutAttr convolution_layout, ::mlir::IntegerAttr feature_group_count, ::mlir::IntegerAttr batch_group_count, ::mlir::ArrayAttr operand_constraints) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::Value weight, ::mlir::Value bias, ::mlir::Value output, ::mlir::DenseI64ArrayAttr window_strides, ::mlir::DenseI64ArrayAttr padding, ::mlir::DenseI64ArrayAttr input_dilation, ::mlir::DenseI64ArrayAttr weight_dilation, ::mlir::DenseBoolArrayAttr window_reversal, ::mlir::tt::ttir::ConvolutionLayoutAttr convolution_layout, ::mlir::IntegerAttr feature_group_count, ::mlir::IntegerAttr batch_group_count, ::mlir::ArrayAttr operand_constraints) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type resultType0, ::mlir::Value input, ::mlir::Value weight, ::mlir::Value bias, ::mlir::Value output, ::llvm::ArrayRef< int64_t > window_strides, ::llvm::ArrayRef< int64_t > padding, ::llvm::ArrayRef< int64_t > input_dilation, ::llvm::ArrayRef< int64_t > weight_dilation, ::llvm::ArrayRef< bool > window_reversal, ::mlir::tt::ttir::ConvolutionLayoutAttr convolution_layout, uint64_t feature_group_count, uint64_t batch_group_count, ::mlir::ArrayAttr operand_constraints) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::Value weight, ::mlir::Value bias, ::mlir::Value output, ::llvm::ArrayRef< int64_t > window_strides, ::llvm::ArrayRef< int64_t > padding, ::llvm::ArrayRef< int64_t > input_dilation, ::llvm::ArrayRef< int64_t > weight_dilation, ::llvm::ArrayRef< bool > window_reversal, ::mlir::tt::ttir::ConvolutionLayoutAttr convolution_layout, uint64_t feature_group_count, uint64_t batch_group_count, ::mlir::ArrayAttr operand_constraints) |
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::ConvolutionOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttir::ConvolutionOp::GenericAdaptor = ConvolutionOpGenericAdaptor<RangeT> |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
uint64_t mlir::tt::ttir::ConvolutionOp::getBatchGroupCount | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
mlir::MutableOperandRange mlir::tt::ttir::ConvolutionOp::getBiasMutable | ( | ) |
mlir::tt::ttir::ConvolutionLayoutAttr mlir::tt::ttir::ConvolutionOp::getConvolutionLayout | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
uint64_t mlir::tt::ttir::ConvolutionOp::getFeatureGroupCount | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
llvm::ArrayRef< int64_t > mlir::tt::ttir::ConvolutionOp::getInputDilation | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
std::pair< unsigned, unsigned > mlir::tt::ttir::ConvolutionOp::getODSOperandIndexAndLength | ( | unsigned | index | ) |
|
inline |
|
inline |
|
inline |
mlir::ArrayAttr mlir::tt::ttir::ConvolutionOp::getOperandConstraints | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inlinestatic |
|
inline |
|
inline |
llvm::ArrayRef< int64_t > mlir::tt::ttir::ConvolutionOp::getPadding | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
llvm::ArrayRef< int64_t > mlir::tt::ttir::ConvolutionOp::getWeightDilation | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
llvm::ArrayRef< bool > mlir::tt::ttir::ConvolutionOp::getWindowReversal | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
llvm::ArrayRef< int64_t > mlir::tt::ttir::ConvolutionOp::getWindowStrides | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
static |
void mlir::tt::ttir::ConvolutionOp::setBatchGroupCount | ( | uint64_t | attrValue | ) |
|
inline |
|
inline |
void mlir::tt::ttir::ConvolutionOp::setFeatureGroupCount | ( | uint64_t | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttir::ConvolutionOp::setInputDilation | ( | ::llvm::ArrayRef< int64_t > | attrValue | ) |
|
inline |
|
inline |
void mlir::tt::ttir::ConvolutionOp::setPadding | ( | ::llvm::ArrayRef< int64_t > | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttir::ConvolutionOp::setWeightDilation | ( | ::llvm::ArrayRef< int64_t > | attrValue | ) |
|
inline |
void mlir::tt::ttir::ConvolutionOp::setWindowReversal | ( | ::llvm::ArrayRef< bool > | attrValue | ) |
|
inline |
void mlir::tt::ttir::ConvolutionOp::setWindowStrides | ( | ::llvm::ArrayRef< int64_t > | attrValue | ) |
|
inline |
::llvm::LogicalResult mlir::tt::ttir::ConvolutionOp::verify | ( | ) |
|
static |
llvm::LogicalResult mlir::tt::ttir::ConvolutionOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttir::ConvolutionOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttir::ConvolutionOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |