mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT > Class Template Reference

Inheritance diagram for mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >:

Public Member Functions
	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 Properties &	getProperties ()
::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::MemoryConfigAttr >	getMemoryConfig ()

Private Types
using	ValueT = ::llvm::detail::ValueOfRange< RangeT >
using	Base = detail::BatchNormOpGenericAdaptorBase

Private Attributes
RangeT	odsOperands

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

Member Typedef Documentation

Base

template<typename RangeT >

using mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::Base = detail::BatchNormOpGenericAdaptorBase

private

ValueT

template<typename RangeT >

using mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::ValueT = ::llvm::detail::ValueOfRange<RangeT>

private

Constructor & Destructor Documentation

BatchNormOpGenericAdaptor() [1/5]

template<typename RangeT >

mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::BatchNormOpGenericAdaptor ( RangeT  values, ::mlir::DictionaryAttr  attrs, const Properties &  properties, ::mlir::RegionRange  regions = {}  )

inline

BatchNormOpGenericAdaptor() [2/5]

template<typename RangeT >

mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::BatchNormOpGenericAdaptor ( RangeT  values, ::mlir::DictionaryAttr  attrs, ::mlir::OpaqueProperties  properties, ::mlir::RegionRange  regions = {}  )

inline

BatchNormOpGenericAdaptor() [3/5]

template<typename RangeT >

mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::BatchNormOpGenericAdaptor ( RangeT  values, ::mlir::DictionaryAttr  attrs  )

inline

BatchNormOpGenericAdaptor() [4/5]

template<typename RangeT >

mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::BatchNormOpGenericAdaptor ( RangeT  values, const BatchNormOpGenericAdaptorBase &  base  )

inline

BatchNormOpGenericAdaptor() [5/5]

template<typename RangeT >

template<typename LateInst = BatchNormOp, typename = std::enable_if_t<std::is_same_v<LateInst, BatchNormOp>>>

mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::BatchNormOpGenericAdaptor ( RangeT  values, LateInst  op  )

inline

Member Function Documentation

getBias()

template<typename RangeT >

ValueT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getBias ( )

inline

getInput()

template<typename RangeT >

ValueT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getInput ( )

inline

getODSOperandIndexAndLength()

template<typename RangeT >

std::pair<unsigned, unsigned> mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getODSOperandIndexAndLength ( unsigned  index )

inline

getODSOperands()

template<typename RangeT >

RangeT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getODSOperands ( unsigned  index )

inline

getOperands()

template<typename RangeT >

RangeT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getOperands ( )

inline

getRunningMean()

template<typename RangeT >

ValueT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getRunningMean ( )

inline

getRunningVar()

template<typename RangeT >

ValueT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getRunningVar ( )

inline

getWeight()

template<typename RangeT >

ValueT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::getWeight ( )

inline

Member Data Documentation

odsOperands

template<typename RangeT >

RangeT mlir::tt::ttnn::BatchNormOpGenericAdaptor< RangeT >::odsOperands

private

---

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

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTNN/IR/TTNNOps.h.inc