mlir::tt::ttir::BatchNormOp Class Reference

Inheritance diagram for mlir::tt::ttir::BatchNormOp:

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)

Member Typedef Documentation

Adaptor

using mlir::tt::ttir::BatchNormOp::Adaptor = BatchNormOpAdaptor

FoldAdaptor

using mlir::tt::ttir::BatchNormOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> >

GenericAdaptor

template<typename RangeT >

using mlir::tt::ttir::BatchNormOp::GenericAdaptor = BatchNormOpGenericAdaptor<RangeT>

Properties

using mlir::tt::ttir::BatchNormOp::Properties = FoldAdaptor::Properties

Member Function Documentation

build() [1/6]

void mlir::tt::ttir::BatchNormOp::build ( ::mlir::OpBuilder &  , ::mlir::OperationState &  odsState, ::mlir::TypeRange  resultTypes, ::mlir::ValueRange  operands, ::llvm::ArrayRef<::mlir::NamedAttribute >  attributes = {}  )

static

build() [2/6]

void mlir::tt::ttir::BatchNormOp::build ( ::mlir::OpBuilder &  , ::mlir::OperationState &  odsState, ::mlir::TypeRange  resultTypes, ::mlir::ValueRange  operands, const Properties &  properties, ::llvm::ArrayRef<::mlir::NamedAttribute >  discardableAttributes = {}  )

static

build() [3/6]

void mlir::tt::ttir::BatchNormOp::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

build() [4/6]

void mlir::tt::ttir::BatchNormOp::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

build() [5/6]

void mlir::tt::ttir::BatchNormOp::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

build() [6/6]

void mlir::tt::ttir::BatchNormOp::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

computePropertiesHash()

llvm::hash_code mlir::tt::ttir::BatchNormOp::computePropertiesHash ( const Properties &  prop )

static

getAttributeNameForIndex() [1/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getAttributeNameForIndex ( ::mlir::OperationName  name, unsigned  index  )

inlinestaticprivate

getAttributeNameForIndex() [2/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getAttributeNameForIndex ( unsigned  index )

inlineprivate

getAttributeNames()

static ::llvm::ArrayRef<::llvm::StringRef> mlir::tt::ttir::BatchNormOp::getAttributeNames ( )

inline

getDimension()

uint32_t mlir::tt::ttir::BatchNormOp::getDimension

(

)

getDimensionAttr()

::mlir::IntegerAttr mlir::tt::ttir::BatchNormOp::getDimensionAttr ( )

inline

getDimensionAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getDimensionAttrName ( )

inline

getDimensionAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getDimensionAttrName ( ::mlir::OperationName  name )

inlinestatic

getDpsInitsMutable()

MutableOperandRange mlir::tt::ttir::BatchNormOp::getDpsInitsMutable ( )

inline

getEpsilon()

llvm::APFloat mlir::tt::ttir::BatchNormOp::getEpsilon

(

)

getEpsilonAttr()

::mlir::FloatAttr mlir::tt::ttir::BatchNormOp::getEpsilonAttr ( )

inline

getEpsilonAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getEpsilonAttrName ( )

inline

getEpsilonAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getEpsilonAttrName ( ::mlir::OperationName  name )

inlinestatic

getInherentAttr()

std::optional< mlir::Attribute > mlir::tt::ttir::BatchNormOp::getInherentAttr ( ::mlir::MLIRContext *  ctx, const Properties &  prop, llvm::StringRef  name  )

static

getMean()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::BatchNormOp::getMean ( )

inline

getMeanMutable()

::mlir::OpOperand& mlir::tt::ttir::BatchNormOp::getMeanMutable ( )

inline

getODSOperandIndexAndLength()

std::pair<unsigned, unsigned> mlir::tt::ttir::BatchNormOp::getODSOperandIndexAndLength ( unsigned  index )

inline

getODSOperands()

::mlir::Operation::operand_range mlir::tt::ttir::BatchNormOp::getODSOperands ( unsigned  index )

inline

getODSResultIndexAndLength()

std::pair<unsigned, unsigned> mlir::tt::ttir::BatchNormOp::getODSResultIndexAndLength ( unsigned  index )

inline

getODSResults()

::mlir::Operation::result_range mlir::tt::ttir::BatchNormOp::getODSResults ( unsigned  index )

inline

getOffset()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::BatchNormOp::getOffset ( )

inline

getOffsetMutable()

::mlir::OpOperand& mlir::tt::ttir::BatchNormOp::getOffsetMutable ( )

inline

getOperand()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::BatchNormOp::getOperand ( )

inline

getOperandMutable()

::mlir::OpOperand& mlir::tt::ttir::BatchNormOp::getOperandMutable ( )

inline

getOperationName()

static constexpr ::llvm::StringLiteral mlir::tt::ttir::BatchNormOp::getOperationName ( )

inlinestatic

getOutput()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::BatchNormOp::getOutput ( )

inline

getOutputMutable()

::mlir::OpOperand& mlir::tt::ttir::BatchNormOp::getOutputMutable ( )

inline

getPropertiesAsAttr()

mlir::Attribute mlir::tt::ttir::BatchNormOp::getPropertiesAsAttr ( ::mlir::MLIRContext *  ctx, const Properties &  prop  )

static

getResult()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::BatchNormOp::getResult ( )

inline

getScale()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::BatchNormOp::getScale ( )

inline

getScaleMutable()

::mlir::OpOperand& mlir::tt::ttir::BatchNormOp::getScaleMutable ( )

inline

getTraining()

bool mlir::tt::ttir::BatchNormOp::getTraining

(

)

getTrainingAttr()

::mlir::BoolAttr mlir::tt::ttir::BatchNormOp::getTrainingAttr ( )

inline

getTrainingAttrName() [1/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getTrainingAttrName ( )

inline

getTrainingAttrName() [2/2]

::mlir::StringAttr mlir::tt::ttir::BatchNormOp::getTrainingAttrName ( ::mlir::OperationName  name )

inlinestatic

getVariance()

::mlir::TypedValue<::mlir::RankedTensorType> mlir::tt::ttir::BatchNormOp::getVariance ( )

inline

getVarianceMutable()

::mlir::OpOperand& mlir::tt::ttir::BatchNormOp::getVarianceMutable ( )

inline

populateInherentAttrs()

void mlir::tt::ttir::BatchNormOp::populateInherentAttrs ( ::mlir::MLIRContext *  ctx, const Properties &  prop, ::mlir::NamedAttrList &  attrs  )

static

readProperties()

llvm::LogicalResult mlir::tt::ttir::BatchNormOp::readProperties ( ::mlir::DialectBytecodeReader &  reader, ::mlir::OperationState &  state  )

static

setDimension()

void mlir::tt::ttir::BatchNormOp::setDimension

(

uint32_t

attrValue

)

setDimensionAttr()

void mlir::tt::ttir::BatchNormOp::setDimensionAttr ( ::mlir::IntegerAttr  attr )

inline

setEpsilon()

void mlir::tt::ttir::BatchNormOp::setEpsilon

(

::llvm::APFloat

attrValue

)

setEpsilonAttr()

void mlir::tt::ttir::BatchNormOp::setEpsilonAttr ( ::mlir::FloatAttr  attr )

inline

setInherentAttr()

void mlir::tt::ttir::BatchNormOp::setInherentAttr ( Properties &  prop, llvm::StringRef  name, mlir::Attribute  value  )

static

setPropertiesFromAttr()

llvm::LogicalResult mlir::tt::ttir::BatchNormOp::setPropertiesFromAttr ( Properties &  prop, ::mlir::Attribute  attr, ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError  )

static

setTraining()

void mlir::tt::ttir::BatchNormOp::setTraining

(

bool

attrValue

)

setTrainingAttr()

void mlir::tt::ttir::BatchNormOp::setTrainingAttr ( ::mlir::BoolAttr  attr )

inline

verify()

::llvm::LogicalResult mlir::tt::ttir::BatchNormOp::verify

(

)

verifyInherentAttrs()

llvm::LogicalResult mlir::tt::ttir::BatchNormOp::verifyInherentAttrs ( ::mlir::OperationName  opName, ::mlir::NamedAttrList &  attrs, llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError  )

static

verifyInvariants()

llvm::LogicalResult mlir::tt::ttir::BatchNormOp::verifyInvariants

(

)

verifyInvariantsImpl()

llvm::LogicalResult mlir::tt::ttir::BatchNormOp::verifyInvariantsImpl

(

)

writeProperties()

void mlir::tt::ttir::BatchNormOp::writeProperties

(

::mlir::DialectBytecodeWriter &

writer

)

---

The documentation for this class was generated from the following files:

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTIR/IR/TTIROps.h.inc
• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTIR/IR/TTIROps.cpp.inc