TT-MLIR
Classes | Public Member Functions | Protected Attributes | List of all members
mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase Class Reference
Inheritance diagram for mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase:
mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT > mlir::tt::ttnn::BatchNormOpAdaptor

Classes

struct  Properties
 

Public Member Functions

 BatchNormOpGenericAdaptorBase (::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 BatchNormOpGenericAdaptorBase (BatchNormOp op)
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index, unsigned odsOperandsSize)
 
const PropertiesgetProperties ()
 
::mlir::DictionaryAttr getAttributes ()
 
::mlir::BoolAttr getTrainingAttr ()
 
bool getTraining ()
 
::mlir::FloatAttr getEpsilonAttr ()
 
::llvm::APFloat getEpsilon ()
 
::mlir::FloatAttr getMomentumAttr ()
 
::llvm::APFloat getMomentum ()
 
::mlir::tt::ttnn::MemoryConfigAttr getMemoryConfigAttr ()
 
::std::optional<::mlir::tt::ttnn::MemoryConfigAttrgetMemoryConfig ()
 

Protected Attributes

::mlir::DictionaryAttr odsAttrs
 
::std::optional<::mlir::OperationName > odsOpName
 
Properties properties
 
::mlir::RegionRange odsRegions
 

Constructor & Destructor Documentation

◆ BatchNormOpGenericAdaptorBase() [1/2]

mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::BatchNormOpGenericAdaptorBase ( ::mlir::DictionaryAttr  attrs,
const Properties properties,
::mlir::RegionRange  regions = {} 
)
inline

◆ BatchNormOpGenericAdaptorBase() [2/2]

mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::BatchNormOpGenericAdaptorBase ( BatchNormOp  op)

Member Function Documentation

◆ getAttributes()

::mlir::DictionaryAttr mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getAttributes ( )
inline

◆ getEpsilon()

llvm::APFloat mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getEpsilon ( )

◆ getEpsilonAttr()

mlir::FloatAttr mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getEpsilonAttr ( )

◆ getMemoryConfig()

std::optional<::mlir::tt::ttnn::MemoryConfigAttr > mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getMemoryConfig ( )

◆ getMemoryConfigAttr()

::mlir::tt::ttnn::MemoryConfigAttr mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getMemoryConfigAttr ( )
inline

◆ getMomentum()

llvm::APFloat mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getMomentum ( )

◆ getMomentumAttr()

mlir::FloatAttr mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getMomentumAttr ( )

◆ getODSOperandIndexAndLength()

std::pair< unsigned, unsigned > mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getODSOperandIndexAndLength ( unsigned  index,
unsigned  odsOperandsSize 
)

◆ getProperties()

const Properties& mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getProperties ( )
inline

◆ getTraining()

bool mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getTraining ( )

◆ getTrainingAttr()

mlir::BoolAttr mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::getTrainingAttr ( )

Member Data Documentation

◆ odsAttrs

::mlir::DictionaryAttr mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::odsAttrs
protected

◆ odsOpName

::std::optional<::mlir::OperationName> mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::odsOpName
protected

◆ odsRegions

::mlir::RegionRange mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::odsRegions
protected

◆ properties

Properties mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase::properties
protected

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