TT-MLIR
|
Public Types | |
using | Adaptor = RequantizeOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = RequantizeOpGenericAdaptor< RangeT > |
using | FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > > |
using | Properties = FoldAdaptor::Properties |
Public Member Functions | |
static ::llvm::ArrayRef<::llvm::StringRef > | getAttributeNames () |
::mlir::StringAttr | getAxisAttrName () |
::mlir::StringAttr | getInScaleAttrName () |
::mlir::StringAttr | getInZeroPointAttrName () |
::mlir::StringAttr | getMemoryConfigAttrName () |
::mlir::StringAttr | getOutScaleAttrName () |
::mlir::StringAttr | getOutZeroPointAttrName () |
::mlir::StringAttr | getOutputDtypeAttrName () |
std::pair< unsigned, unsigned > | getODSOperandIndexAndLength (unsigned index) |
::mlir::Operation::operand_range | getODSOperands (unsigned index) |
::mlir::TypedValue<::mlir::RankedTensorType > | getInput () |
::mlir::OpOperand & | getInputMutable () |
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::FloatAttr | getInScaleAttr () |
::llvm::APFloat | getInScale () |
::mlir::IntegerAttr | getInZeroPointAttr () |
uint32_t | getInZeroPoint () |
::mlir::FloatAttr | getOutScaleAttr () |
::llvm::APFloat | getOutScale () |
::mlir::IntegerAttr | getOutZeroPointAttr () |
uint32_t | getOutZeroPoint () |
::mlir::IntegerAttr | getAxisAttr () |
::std::optional< uint32_t > | getAxis () |
::mlir::tt::DataTypeAttr | getOutputDtypeAttr () |
::std::optional<::mlir::tt::DataType > | getOutputDtype () |
::mlir::tt::ttnn::MemoryConfigAttr | getMemoryConfigAttr () |
::std::optional<::mlir::tt::ttnn::MemoryConfigAttr > | getMemoryConfig () |
void | setInScaleAttr (::mlir::FloatAttr attr) |
void | setInScale (::llvm::APFloat attrValue) |
void | setInZeroPointAttr (::mlir::IntegerAttr attr) |
void | setInZeroPoint (uint32_t attrValue) |
void | setOutScaleAttr (::mlir::FloatAttr attr) |
void | setOutScale (::llvm::APFloat attrValue) |
void | setOutZeroPointAttr (::mlir::IntegerAttr attr) |
void | setOutZeroPoint (uint32_t attrValue) |
void | setAxisAttr (::mlir::IntegerAttr attr) |
void | setAxis (::std::optional< uint32_t > attrValue) |
void | setOutputDtypeAttr (::mlir::tt::DataTypeAttr attr) |
void | setOutputDtype (::std::optional<::mlir::tt::DataType > attrValue) |
void | setMemoryConfigAttr (::mlir::tt::ttnn::MemoryConfigAttr attr) |
::mlir::Attribute | removeAxisAttr () |
::mlir::Attribute | removeOutputDtypeAttr () |
::mlir::Attribute | removeMemoryConfigAttr () |
::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 | getAxisAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInScaleAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getInZeroPointAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getMemoryConfigAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getOutScaleAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getOutZeroPointAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getOutputDtypeAttrName (::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::FloatAttr in_scale, ::mlir::IntegerAttr in_zero_point, ::mlir::FloatAttr out_scale, ::mlir::IntegerAttr out_zero_point, ::mlir::IntegerAttr axis, ::mlir::tt::DataTypeAttr output_dtype, ::mlir::tt::ttnn::MemoryConfigAttr memory_config) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::mlir::FloatAttr in_scale, ::mlir::IntegerAttr in_zero_point, ::mlir::FloatAttr out_scale, ::mlir::IntegerAttr out_zero_point, ::mlir::IntegerAttr axis, ::mlir::tt::DataTypeAttr output_dtype, ::mlir::tt::ttnn::MemoryConfigAttr memory_config) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value input, ::llvm::APFloat in_scale, uint32_t in_zero_point, ::llvm::APFloat out_scale, uint32_t out_zero_point, ::mlir::IntegerAttr axis, ::mlir::tt::DataTypeAttr output_dtype, ::mlir::tt::ttnn::MemoryConfigAttr memory_config) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value input, ::llvm::APFloat in_scale, uint32_t in_zero_point, ::llvm::APFloat out_scale, uint32_t out_zero_point, ::mlir::IntegerAttr axis, ::mlir::tt::DataTypeAttr output_dtype, ::mlir::tt::ttnn::MemoryConfigAttr memory_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::RequantizeOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttnn::RequantizeOp::GenericAdaptor = RequantizeOpGenericAdaptor<RangeT> |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
std::optional< uint32_t > mlir::tt::ttnn::RequantizeOp::getAxis | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
void mlir::tt::ttnn::RequantizeOp::getEffects | ( | ::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> & | effects | ) |
|
static |
|
inline |
|
inline |
llvm::APFloat mlir::tt::ttnn::RequantizeOp::getInScale | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::RequantizeOp::getInZeroPoint | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
std::optional<::mlir::tt::ttnn::MemoryConfigAttr > mlir::tt::ttnn::RequantizeOp::getMemoryConfig | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
inline |
|
inline |
|
inlinestatic |
std::optional<::mlir::tt::DataType > mlir::tt::ttnn::RequantizeOp::getOutputDtype | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
llvm::APFloat mlir::tt::ttnn::RequantizeOp::getOutScale | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
uint32_t mlir::tt::ttnn::RequantizeOp::getOutZeroPoint | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
|
static |
|
static |
|
inline |
|
inline |
|
inline |
void mlir::tt::ttnn::RequantizeOp::setAxis | ( | ::std::optional< uint32_t > | attrValue | ) |
|
inline |
|
static |
void mlir::tt::ttnn::RequantizeOp::setInScale | ( | ::llvm::APFloat | attrValue | ) |
|
inline |
void mlir::tt::ttnn::RequantizeOp::setInZeroPoint | ( | uint32_t | attrValue | ) |
|
inline |
|
inline |
void mlir::tt::ttnn::RequantizeOp::setOutputDtype | ( | ::std::optional<::mlir::tt::DataType > | attrValue | ) |
|
inline |
void mlir::tt::ttnn::RequantizeOp::setOutScale | ( | ::llvm::APFloat | attrValue | ) |
|
inline |
void mlir::tt::ttnn::RequantizeOp::setOutZeroPoint | ( | uint32_t | attrValue | ) |
|
inline |
|
static |
::llvm::LogicalResult mlir::tt::ttnn::RequantizeOp::verify | ( | ) |
|
static |
llvm::LogicalResult mlir::tt::ttnn::RequantizeOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttnn::RequantizeOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttnn::RequantizeOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |