TT-MLIR
|
Public Types | |
using | Adaptor = BatchNormOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = BatchNormOpGenericAdaptor< RangeT > |
using | FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > > |
using | Properties = FoldAdaptor::Properties |
Public Member Functions | |
static ::llvm::ArrayRef<::llvm::StringRef > | getAttributeNames () |
::mlir::StringAttr | getDimensionAttrName () |
::mlir::StringAttr | getEpsilonAttrName () |
::mlir::StringAttr | getTrainingAttrName () |
std::pair< unsigned, unsigned > | getODSOperandIndexAndLength (unsigned index) |
::mlir::Operation::operand_range | getODSOperands (unsigned index) |
::mlir::TypedValue<::mlir::RankedTensorType > | getOperand () |
::mlir::TypedValue<::mlir::RankedTensorType > | getScale () |
::mlir::TypedValue<::mlir::RankedTensorType > | getOffset () |
::mlir::TypedValue<::mlir::RankedTensorType > | getMean () |
::mlir::TypedValue<::mlir::RankedTensorType > | getVariance () |
::mlir::TypedValue<::mlir::RankedTensorType > | getOutput () |
::mlir::OpOperand & | getOperandMutable () |
::mlir::OpOperand & | getScaleMutable () |
::mlir::OpOperand & | getOffsetMutable () |
::mlir::OpOperand & | getMeanMutable () |
::mlir::OpOperand & | getVarianceMutable () |
::mlir::OpOperand & | getOutputMutable () |
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 | getEpsilonAttr () |
::llvm::APFloat | getEpsilon () |
::mlir::IntegerAttr | getDimensionAttr () |
uint32_t | getDimension () |
::mlir::BoolAttr | getTrainingAttr () |
bool | getTraining () |
void | setEpsilonAttr (::mlir::FloatAttr attr) |
void | setEpsilon (::llvm::APFloat attrValue) |
void | setDimensionAttr (::mlir::IntegerAttr attr) |
void | setDimension (uint32_t attrValue) |
void | setTrainingAttr (::mlir::BoolAttr attr) |
void | setTraining (bool attrValue) |
::llvm::LogicalResult | verifyInvariantsImpl () |
::llvm::LogicalResult | verifyInvariants () |
::llvm::LogicalResult | verify () |
MutableOperandRange | getDpsInitsMutable () |
Static Public Member Functions | |
::mlir::StringAttr | getDimensionAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getEpsilonAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getTrainingAttrName (::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 operand, ::mlir::Value scale, ::mlir::Value offset, ::mlir::Value mean, ::mlir::Value variance, ::mlir::Value output, ::mlir::FloatAttr epsilon, ::mlir::IntegerAttr dimension, ::mlir::BoolAttr training) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value operand, ::mlir::Value scale, ::mlir::Value offset, ::mlir::Value mean, ::mlir::Value variance, ::mlir::Value output, ::mlir::FloatAttr epsilon, ::mlir::IntegerAttr dimension, ::mlir::BoolAttr training) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value operand, ::mlir::Value scale, ::mlir::Value offset, ::mlir::Value mean, ::mlir::Value variance, ::mlir::Value output, ::llvm::APFloat epsilon, uint32_t dimension, bool training) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value operand, ::mlir::Value scale, ::mlir::Value offset, ::mlir::Value mean, ::mlir::Value variance, ::mlir::Value output, ::llvm::APFloat epsilon, uint32_t dimension, bool training) |
static void | build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute > attributes={}) |
static void | build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, const Properties &properties, ::llvm::ArrayRef<::mlir::NamedAttribute > discardableAttributes={}) |
Private Member Functions | |
::mlir::StringAttr | getAttributeNameForIndex (unsigned index) |
Static Private Member Functions | |
::mlir::StringAttr | getAttributeNameForIndex (::mlir::OperationName name, unsigned index) |
using mlir::tt::ttir::BatchNormOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttir::BatchNormOp::GenericAdaptor = BatchNormOpGenericAdaptor<RangeT> |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
uint32_t mlir::tt::ttir::BatchNormOp::getDimension | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
llvm::APFloat mlir::tt::ttir::BatchNormOp::getEpsilon | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
static |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
static |
|
inline |
|
inline |
|
inline |
bool mlir::tt::ttir::BatchNormOp::getTraining | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
static |
|
static |
void mlir::tt::ttir::BatchNormOp::setDimension | ( | uint32_t | attrValue | ) |
|
inline |
void mlir::tt::ttir::BatchNormOp::setEpsilon | ( | ::llvm::APFloat | attrValue | ) |
|
inline |
|
static |
|
static |
void mlir::tt::ttir::BatchNormOp::setTraining | ( | bool | attrValue | ) |
|
inline |
::llvm::LogicalResult mlir::tt::ttir::BatchNormOp::verify | ( | ) |
|
static |
llvm::LogicalResult mlir::tt::ttir::BatchNormOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttir::BatchNormOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttir::BatchNormOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |