TT-MLIR
|
Public Types | |
using | Adaptor = Conv2dOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = Conv2dOpGenericAdaptor< RangeT > |
using | FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > > |
using | Properties = FoldAdaptor::Properties |
Public Member Functions | |
static ::llvm::ArrayRef<::llvm::StringRef > | getAttributeNames () |
::mlir::StringAttr | getBatchSizeAttrName () |
::mlir::StringAttr | getDilationHeightAttrName () |
::mlir::StringAttr | getDilationWidthAttrName () |
::mlir::StringAttr | getGroupsAttrName () |
::mlir::StringAttr | getInChannelsAttrName () |
::mlir::StringAttr | getInputHeightAttrName () |
::mlir::StringAttr | getInputWidthAttrName () |
::mlir::StringAttr | getKernelHeightAttrName () |
::mlir::StringAttr | getKernelWidthAttrName () |
::mlir::StringAttr | getOutChannelsAttrName () |
::mlir::StringAttr | getPaddingHeightAttrName () |
::mlir::StringAttr | getPaddingWidthAttrName () |
::mlir::StringAttr | getStrideHeightAttrName () |
::mlir::StringAttr | getStrideWidthAttrName () |
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::TypedValue<::mlir::tt::DeviceType > | getDevice () |
::mlir::OpOperand & | getInputMutable () |
::mlir::OpOperand & | getWeightMutable () |
::mlir::MutableOperandRange | getBiasMutable () |
::mlir::OpOperand & | getOutputMutable () |
::mlir::OpOperand & | getDeviceMutable () |
std::pair< unsigned, unsigned > | getODSResultIndexAndLength (unsigned index) |
::mlir::Operation::result_range | getODSResults (unsigned index) |
::mlir::TypedValue<::mlir::RankedTensorType > | getResult () |
void | writeProperties (::mlir::DialectBytecodeWriter &writer) |
::mlir::IntegerAttr | getInChannelsAttr () |
uint32_t | getInChannels () |
::mlir::IntegerAttr | getOutChannelsAttr () |
uint32_t | getOutChannels () |
::mlir::IntegerAttr | getBatchSizeAttr () |
uint32_t | getBatchSize () |
::mlir::IntegerAttr | getInputHeightAttr () |
uint32_t | getInputHeight () |
::mlir::IntegerAttr | getInputWidthAttr () |
uint32_t | getInputWidth () |
::mlir::IntegerAttr | getKernelHeightAttr () |
uint32_t | getKernelHeight () |
::mlir::IntegerAttr | getKernelWidthAttr () |
uint32_t | getKernelWidth () |
::mlir::IntegerAttr | getStrideHeightAttr () |
uint32_t | getStrideHeight () |
::mlir::IntegerAttr | getStrideWidthAttr () |
uint32_t | getStrideWidth () |
::mlir::IntegerAttr | getPaddingHeightAttr () |
uint32_t | getPaddingHeight () |
::mlir::IntegerAttr | getPaddingWidthAttr () |
uint32_t | getPaddingWidth () |
::mlir::IntegerAttr | getDilationHeightAttr () |
uint32_t | getDilationHeight () |
::mlir::IntegerAttr | getDilationWidthAttr () |
uint32_t | getDilationWidth () |
::mlir::IntegerAttr | getGroupsAttr () |
uint32_t | getGroups () |
void | setInChannelsAttr (::mlir::IntegerAttr attr) |
void | setInChannels (uint32_t attrValue) |
void | setOutChannelsAttr (::mlir::IntegerAttr attr) |
void | setOutChannels (uint32_t attrValue) |
void | setBatchSizeAttr (::mlir::IntegerAttr attr) |
void | setBatchSize (uint32_t attrValue) |
void | setInputHeightAttr (::mlir::IntegerAttr attr) |
void | setInputHeight (uint32_t attrValue) |
void | setInputWidthAttr (::mlir::IntegerAttr attr) |
void | setInputWidth (uint32_t attrValue) |
void | setKernelHeightAttr (::mlir::IntegerAttr attr) |
void | setKernelHeight (uint32_t attrValue) |
void | setKernelWidthAttr (::mlir::IntegerAttr attr) |
void | setKernelWidth (uint32_t attrValue) |
void | setStrideHeightAttr (::mlir::IntegerAttr attr) |
void | setStrideHeight (uint32_t attrValue) |
void | setStrideWidthAttr (::mlir::IntegerAttr attr) |
void | setStrideWidth (uint32_t attrValue) |
void | setPaddingHeightAttr (::mlir::IntegerAttr attr) |
void | setPaddingHeight (uint32_t attrValue) |
void | setPaddingWidthAttr (::mlir::IntegerAttr attr) |
void | setPaddingWidth (uint32_t attrValue) |
void | setDilationHeightAttr (::mlir::IntegerAttr attr) |
void | setDilationHeight (uint32_t attrValue) |
void | setDilationWidthAttr (::mlir::IntegerAttr attr) |
void | setDilationWidth (uint32_t attrValue) |
void | setGroupsAttr (::mlir::IntegerAttr attr) |
void | setGroups (uint32_t attrValue) |
::llvm::LogicalResult | verifyInvariantsImpl () |
::llvm::LogicalResult | verifyInvariants () |
::llvm::LogicalResult | verify () |
MutableOperandRange | getDpsInitsMutable () |
Static Public Member Functions | |
::mlir::StringAttr | getBatchSizeAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getDilationHeightAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getDilationWidthAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getGroupsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInChannelsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputHeightAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputWidthAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getKernelHeightAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getKernelWidthAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getOutChannelsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getPaddingHeightAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getPaddingWidthAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getStrideHeightAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getStrideWidthAttrName (::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 weight, ::mlir::Value bias, ::mlir::Value output, ::mlir::Value device, ::mlir::IntegerAttr in_channels, ::mlir::IntegerAttr out_channels, ::mlir::IntegerAttr batch_size, ::mlir::IntegerAttr input_height, ::mlir::IntegerAttr input_width, ::mlir::IntegerAttr kernel_height, ::mlir::IntegerAttr kernel_width, ::mlir::IntegerAttr stride_height, ::mlir::IntegerAttr stride_width, ::mlir::IntegerAttr padding_height, ::mlir::IntegerAttr padding_width, ::mlir::IntegerAttr dilation_height, ::mlir::IntegerAttr dilation_width, ::mlir::IntegerAttr groups) |
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::Value device, ::mlir::IntegerAttr in_channels, ::mlir::IntegerAttr out_channels, ::mlir::IntegerAttr batch_size, ::mlir::IntegerAttr input_height, ::mlir::IntegerAttr input_width, ::mlir::IntegerAttr kernel_height, ::mlir::IntegerAttr kernel_width, ::mlir::IntegerAttr stride_height, ::mlir::IntegerAttr stride_width, ::mlir::IntegerAttr padding_height, ::mlir::IntegerAttr padding_width, ::mlir::IntegerAttr dilation_height, ::mlir::IntegerAttr dilation_width, ::mlir::IntegerAttr groups) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value input, ::mlir::Value weight, ::mlir::Value bias, ::mlir::Value output, ::mlir::Value device, uint32_t in_channels, uint32_t out_channels, uint32_t batch_size, uint32_t input_height, uint32_t input_width, uint32_t kernel_height, uint32_t kernel_width, uint32_t stride_height, uint32_t stride_width, uint32_t padding_height, uint32_t padding_width, uint32_t dilation_height, uint32_t dilation_width, uint32_t groups) |
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::Value device, uint32_t in_channels, uint32_t out_channels, uint32_t batch_size, uint32_t input_height, uint32_t input_width, uint32_t kernel_height, uint32_t kernel_width, uint32_t stride_height, uint32_t stride_width, uint32_t padding_height, uint32_t padding_width, uint32_t dilation_height, uint32_t dilation_width, uint32_t groups) |
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::ttnn::Conv2dOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttnn::Conv2dOp::GenericAdaptor = Conv2dOpGenericAdaptor<RangeT> |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
uint32_t mlir::tt::ttnn::Conv2dOp::getBatchSize | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
mlir::MutableOperandRange mlir::tt::ttnn::Conv2dOp::getBiasMutable | ( | ) |
|
inline |
|
inline |
uint32_t mlir::tt::ttnn::Conv2dOp::getDilationHeight | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::Conv2dOp::getDilationWidth | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
uint32_t mlir::tt::ttnn::Conv2dOp::getGroups | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::Conv2dOp::getInChannels | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
uint32_t mlir::tt::ttnn::Conv2dOp::getInputHeight | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
uint32_t mlir::tt::ttnn::Conv2dOp::getInputWidth | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::Conv2dOp::getKernelHeight | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::Conv2dOp::getKernelWidth | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
std::pair< unsigned, unsigned > mlir::tt::ttnn::Conv2dOp::getODSOperandIndexAndLength | ( | unsigned | index | ) |
|
inline |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::Conv2dOp::getOutChannels | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
uint32_t mlir::tt::ttnn::Conv2dOp::getPaddingHeight | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::Conv2dOp::getPaddingWidth | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
uint32_t mlir::tt::ttnn::Conv2dOp::getStrideHeight | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::Conv2dOp::getStrideWidth | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
static |
|
static |
void mlir::tt::ttnn::Conv2dOp::setBatchSize | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setDilationHeight | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setDilationWidth | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setGroups | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setInChannels | ( | uint32_t | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttnn::Conv2dOp::setInputHeight | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setInputWidth | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setKernelHeight | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setKernelWidth | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setOutChannels | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setPaddingHeight | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setPaddingWidth | ( | uint32_t | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttnn::Conv2dOp::setStrideHeight | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::Conv2dOp::setStrideWidth | ( | uint32_t | attrValue | ) |
|
inline |
::llvm::LogicalResult mlir::tt::ttnn::Conv2dOp::verify | ( | ) |
|
static |
llvm::LogicalResult mlir::tt::ttnn::Conv2dOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttnn::Conv2dOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttnn::Conv2dOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |