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

Public Member Functions

 BatchNormOpAdaptor (BatchNormOp op)
 
::llvm::LogicalResult verify (::mlir::Location loc)
 
 BatchNormOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 BatchNormOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions={})
 
 BatchNormOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs)
 
 BatchNormOpGenericAdaptor (RangeT values, const BatchNormOpGenericAdaptorBase &base)
 
template<typename LateInst = BatchNormOp, typename = std::enable_if_t<std::is_same_v<LateInst, BatchNormOp>>>
 BatchNormOpGenericAdaptor (RangeT values, LateInst op)
 
- Public Member Functions inherited from mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >
 BatchNormOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 BatchNormOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions={})
 
 BatchNormOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs)
 
 BatchNormOpGenericAdaptor (RangeT values, const BatchNormOpGenericAdaptorBase &base)
 
template<typename LateInst = BatchNormOp, typename = std::enable_if_t<std::is_same_v<LateInst, BatchNormOp>>>
 BatchNormOpGenericAdaptor (RangeT values, LateInst op)
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index)
 
RangeT getODSOperands (unsigned index)
 
ValueT getInput ()
 
ValueT getRunningMean ()
 
ValueT getRunningVar ()
 
ValueT getWeight ()
 
ValueT getBias ()
 
RangeT getOperands ()
 
- Public Member Functions inherited from mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase
 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 ()
 

Additional Inherited Members

- Protected Attributes inherited from mlir::tt::ttnn::detail::BatchNormOpGenericAdaptorBase
::mlir::DictionaryAttr odsAttrs
 
::std::optional<::mlir::OperationName > odsOpName
 
Properties properties
 
::mlir::RegionRange odsRegions
 

Constructor & Destructor Documentation

◆ BatchNormOpAdaptor()

mlir::tt::ttnn::BatchNormOpAdaptor::BatchNormOpAdaptor ( BatchNormOp  op)

Member Function Documentation

◆ BatchNormOpGenericAdaptor() [1/5]

◆ BatchNormOpGenericAdaptor() [2/5]

◆ BatchNormOpGenericAdaptor() [3/5]

◆ BatchNormOpGenericAdaptor() [4/5]

◆ BatchNormOpGenericAdaptor() [5/5]

template<typename LateInst = BatchNormOp, typename = std::enable_if_t<std::is_same_v<LateInst, BatchNormOp>>>
mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::BatchNormOpGenericAdaptor ( typename LateInst  = BatchNormOp,
typename  = std::enable_if_t<std::is_same_v<LateInst, BatchNormOp>> 
)
inline

◆ verify()

llvm::LogicalResult mlir::tt::ttnn::BatchNormOpAdaptor::verify ( ::mlir::Location  loc)

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