TT-MLIR
|
Public Types | |
using | Adaptor = PrepareConv2dWeightsOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = PrepareConv2dWeightsOpGenericAdaptor< 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 | getConv2dConfigAttrName () |
::mlir::StringAttr | getDilationAttrName () |
::mlir::StringAttr | getGroupsAttrName () |
::mlir::StringAttr | getHasBiasAttrName () |
::mlir::StringAttr | getInChannelsAttrName () |
::mlir::StringAttr | getInputHeightAttrName () |
::mlir::StringAttr | getInputMemoryConfigAttrName () |
::mlir::StringAttr | getInputTensorLayoutAttrName () |
::mlir::StringAttr | getInputWidthAttrName () |
::mlir::StringAttr | getKernelSizeAttrName () |
::mlir::StringAttr | getOutChannelsAttrName () |
::mlir::StringAttr | getPaddingAttrName () |
::mlir::StringAttr | getStrideAttrName () |
::mlir::StringAttr | getWeightsFormatAttrName () |
std::pair< unsigned, unsigned > | getODSOperandIndexAndLength (unsigned index) |
::mlir::Operation::operand_range | getODSOperands (unsigned index) |
::mlir::TypedValue<::mlir::RankedTensorType > | getWeightTensor () |
::mlir::TypedValue<::mlir::tt::ttnn::DeviceType > | getDevice () |
::mlir::OpOperand & | getWeightTensorMutable () |
::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::tt::ttnn::MemoryConfigAttr | getInputMemoryConfigAttr () |
::mlir::tt::ttnn::MemoryConfigAttr | getInputMemoryConfig () |
::mlir::tt::ttnn::LayoutAttr | getInputTensorLayoutAttr () |
::mlir::tt::ttnn::Layout | getInputTensorLayout () |
::mlir::StringAttr | getWeightsFormatAttr () |
::llvm::StringRef | getWeightsFormat () |
::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::DenseI32ArrayAttr | getKernelSizeAttr () |
::llvm::ArrayRef< int32_t > | getKernelSize () |
::mlir::DenseI32ArrayAttr | getStrideAttr () |
::llvm::ArrayRef< int32_t > | getStride () |
::mlir::DenseI32ArrayAttr | getPaddingAttr () |
::llvm::ArrayRef< int32_t > | getPadding () |
::mlir::DenseI32ArrayAttr | getDilationAttr () |
::llvm::ArrayRef< int32_t > | getDilation () |
::mlir::BoolAttr | getHasBiasAttr () |
bool | getHasBias () |
::mlir::IntegerAttr | getGroupsAttr () |
uint32_t | getGroups () |
::mlir::tt::ttnn::Conv2dConfigAttr | getConv2dConfigAttr () |
::std::optional<::mlir::tt::ttnn::Conv2dConfigAttr > | getConv2dConfig () |
void | setInputMemoryConfigAttr (::mlir::tt::ttnn::MemoryConfigAttr attr) |
void | setInputTensorLayoutAttr (::mlir::tt::ttnn::LayoutAttr attr) |
void | setInputTensorLayout (::mlir::tt::ttnn::Layout attrValue) |
void | setWeightsFormatAttr (::mlir::StringAttr attr) |
void | setWeightsFormat (::llvm::StringRef attrValue) |
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 | setKernelSizeAttr (::mlir::DenseI32ArrayAttr attr) |
void | setKernelSize (::llvm::ArrayRef< int32_t > attrValue) |
void | setStrideAttr (::mlir::DenseI32ArrayAttr attr) |
void | setStride (::llvm::ArrayRef< int32_t > attrValue) |
void | setPaddingAttr (::mlir::DenseI32ArrayAttr attr) |
void | setPadding (::llvm::ArrayRef< int32_t > attrValue) |
void | setDilationAttr (::mlir::DenseI32ArrayAttr attr) |
void | setDilation (::llvm::ArrayRef< int32_t > attrValue) |
void | setHasBiasAttr (::mlir::BoolAttr attr) |
void | setHasBias (bool attrValue) |
void | setGroupsAttr (::mlir::IntegerAttr attr) |
void | setGroups (uint32_t attrValue) |
void | setConv2dConfigAttr (::mlir::tt::ttnn::Conv2dConfigAttr attr) |
::mlir::Attribute | removeConv2dConfigAttr () |
::llvm::LogicalResult | verifyInvariantsImpl () |
::llvm::LogicalResult | verifyInvariants () |
::llvm::LogicalResult | verify () |
void | getEffects (::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> &effects) |
Static Public Member Functions | |
::mlir::StringAttr | getBatchSizeAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getConv2dConfigAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getDilationAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getGroupsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getHasBiasAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInChannelsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputHeightAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputMemoryConfigAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputTensorLayoutAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInputWidthAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getKernelSizeAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getOutChannelsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getPaddingAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getStrideAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getWeightsFormatAttrName (::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 weight_tensor, ::mlir::tt::ttnn::MemoryConfigAttr input_memory_config, ::mlir::tt::ttnn::LayoutAttr input_tensor_layout, ::mlir::StringAttr weights_format, ::mlir::IntegerAttr in_channels, ::mlir::IntegerAttr out_channels, ::mlir::IntegerAttr batch_size, ::mlir::IntegerAttr input_height, ::mlir::IntegerAttr input_width, ::mlir::DenseI32ArrayAttr kernel_size, ::mlir::DenseI32ArrayAttr stride, ::mlir::DenseI32ArrayAttr padding, ::mlir::DenseI32ArrayAttr dilation, ::mlir::BoolAttr has_bias, ::mlir::IntegerAttr groups, ::mlir::Value device, ::mlir::tt::ttnn::Conv2dConfigAttr conv2d_config) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value weight_tensor, ::mlir::tt::ttnn::MemoryConfigAttr input_memory_config, ::mlir::tt::ttnn::LayoutAttr input_tensor_layout, ::mlir::StringAttr weights_format, ::mlir::IntegerAttr in_channels, ::mlir::IntegerAttr out_channels, ::mlir::IntegerAttr batch_size, ::mlir::IntegerAttr input_height, ::mlir::IntegerAttr input_width, ::mlir::DenseI32ArrayAttr kernel_size, ::mlir::DenseI32ArrayAttr stride, ::mlir::DenseI32ArrayAttr padding, ::mlir::DenseI32ArrayAttr dilation, ::mlir::BoolAttr has_bias, ::mlir::IntegerAttr groups, ::mlir::Value device, ::mlir::tt::ttnn::Conv2dConfigAttr conv2d_config) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value weight_tensor, ::mlir::tt::ttnn::MemoryConfigAttr input_memory_config, ::mlir::tt::ttnn::Layout input_tensor_layout, ::llvm::StringRef weights_format, uint32_t in_channels, uint32_t out_channels, uint32_t batch_size, uint32_t input_height, uint32_t input_width, ::llvm::ArrayRef< int32_t > kernel_size, ::llvm::ArrayRef< int32_t > stride, ::llvm::ArrayRef< int32_t > padding, ::llvm::ArrayRef< int32_t > dilation, bool has_bias, uint32_t groups, ::mlir::Value device, ::mlir::tt::ttnn::Conv2dConfigAttr conv2d_config) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value weight_tensor, ::mlir::tt::ttnn::MemoryConfigAttr input_memory_config, ::mlir::tt::ttnn::Layout input_tensor_layout, ::llvm::StringRef weights_format, uint32_t in_channels, uint32_t out_channels, uint32_t batch_size, uint32_t input_height, uint32_t input_width, ::llvm::ArrayRef< int32_t > kernel_size, ::llvm::ArrayRef< int32_t > stride, ::llvm::ArrayRef< int32_t > padding, ::llvm::ArrayRef< int32_t > dilation, bool has_bias, uint32_t groups, ::mlir::Value device, ::mlir::tt::ttnn::Conv2dConfigAttr conv2d_config) |
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::PrepareConv2dWeightsOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttnn::PrepareConv2dWeightsOp::GenericAdaptor = PrepareConv2dWeightsOpGenericAdaptor<RangeT> |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
uint32_t mlir::tt::ttnn::PrepareConv2dWeightsOp::getBatchSize | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
std::optional<::mlir::tt::ttnn::Conv2dConfigAttr > mlir::tt::ttnn::PrepareConv2dWeightsOp::getConv2dConfig | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
llvm::ArrayRef< int32_t > mlir::tt::ttnn::PrepareConv2dWeightsOp::getDilation | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::getEffects | ( | ::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> & | effects | ) |
uint32_t mlir::tt::ttnn::PrepareConv2dWeightsOp::getGroups | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
bool mlir::tt::ttnn::PrepareConv2dWeightsOp::getHasBias | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::PrepareConv2dWeightsOp::getInChannels | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
uint32_t mlir::tt::ttnn::PrepareConv2dWeightsOp::getInputHeight | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
mlir::tt::ttnn::MemoryConfigAttr mlir::tt::ttnn::PrepareConv2dWeightsOp::getInputMemoryConfig | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
mlir::tt::ttnn::Layout mlir::tt::ttnn::PrepareConv2dWeightsOp::getInputTensorLayout | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::PrepareConv2dWeightsOp::getInputWidth | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
llvm::ArrayRef< int32_t > mlir::tt::ttnn::PrepareConv2dWeightsOp::getKernelSize | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::PrepareConv2dWeightsOp::getOutChannels | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
llvm::ArrayRef< int32_t > mlir::tt::ttnn::PrepareConv2dWeightsOp::getPadding | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
llvm::ArrayRef< int32_t > mlir::tt::ttnn::PrepareConv2dWeightsOp::getStride | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
llvm::StringRef mlir::tt::ttnn::PrepareConv2dWeightsOp::getWeightsFormat | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
static |
|
static |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setBatchSize | ( | uint32_t | attrValue | ) |
|
inline |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setDilation | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setGroups | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setHasBias | ( | bool | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setInChannels | ( | uint32_t | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setInputHeight | ( | uint32_t | attrValue | ) |
|
inline |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setInputTensorLayout | ( | ::mlir::tt::ttnn::Layout | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setInputWidth | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setKernelSize | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setOutChannels | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setPadding | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setStride | ( | ::llvm::ArrayRef< int32_t > | attrValue | ) |
|
inline |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::setWeightsFormat | ( | ::llvm::StringRef | attrValue | ) |
|
inline |
::llvm::LogicalResult mlir::tt::ttnn::PrepareConv2dWeightsOp::verify | ( | ) |
|
static |
llvm::LogicalResult mlir::tt::ttnn::PrepareConv2dWeightsOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttnn::PrepareConv2dWeightsOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttnn::PrepareConv2dWeightsOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |