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

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

Public Member Functions
	AllReduceOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
	AllReduceOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions={})
	AllReduceOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs=nullptr)
template<typename LateInst = AllReduceOp, typename = std::enable_if_t<std::is_same_v<LateInst, AllReduceOp>>>
	AllReduceOpGenericAdaptor (RangeT values, LateInst op)
std::pair< unsigned, unsigned >	getODSOperandIndexAndLength (unsigned index)
RangeT	getODSOperands (unsigned index)
ValueT	getInput ()
ValueT	getDevice ()
RangeT	getOperands ()
Public Member Functions inherited from mlir::tt::ttnn::detail::AllReduceOpGenericAdaptorBase
	AllReduceOpGenericAdaptorBase (::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
	AllReduceOpGenericAdaptorBase (AllReduceOp op)
std::pair< unsigned, unsigned >	getODSOperandIndexAndLength (unsigned index, unsigned odsOperandsSize)
const Properties &	getProperties ()
::mlir::DictionaryAttr	getAttributes ()
::mlir::IntegerAttr	getScatterDimAttr ()
int32_t	getScatterDim ()
::mlir::IntegerAttr	getScatterNumAttr ()
int32_t	getScatterNum ()
::mlir::tt::ReduceTypeAttr	getMathOpAttr ()
::mlir::tt::ReduceType	getMathOp ()
::mlir::IntegerAttr	getNumLinksAttr ()
int32_t	getNumLinks ()

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

Private Attributes
RangeT	odsOperands

Additional Inherited Members
Protected Attributes inherited from mlir::tt::ttnn::detail::AllReduceOpGenericAdaptorBase
::mlir::DictionaryAttr	odsAttrs
::std::optional<::mlir::OperationName >	odsOpName
Properties	properties
::mlir::RegionRange	odsRegions

Member Typedef Documentation

Base

template<typename RangeT >

using mlir::tt::ttnn::AllReduceOpGenericAdaptor< RangeT >::Base = detail::AllReduceOpGenericAdaptorBase

private

ValueT

template<typename RangeT >

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

private

Constructor & Destructor Documentation

AllReduceOpGenericAdaptor() [1/4]

template<typename RangeT >

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

inline

AllReduceOpGenericAdaptor() [2/4]

template<typename RangeT >

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

inline

AllReduceOpGenericAdaptor() [3/4]

template<typename RangeT >

mlir::tt::ttnn::AllReduceOpGenericAdaptor< RangeT >::AllReduceOpGenericAdaptor ( RangeT  values, ::mlir::DictionaryAttr  attrs = nullptr  )

inline

AllReduceOpGenericAdaptor() [4/4]

template<typename RangeT >

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

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

inline

Member Function Documentation

getDevice()

template<typename RangeT >

ValueT mlir::tt::ttnn::AllReduceOpGenericAdaptor< RangeT >::getDevice ( )

inline

getInput()

template<typename RangeT >

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

inline

getODSOperandIndexAndLength()

template<typename RangeT >

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

inline

getODSOperands()

template<typename RangeT >

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

inline

getOperands()

template<typename RangeT >

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

inline

Member Data Documentation

odsOperands

template<typename RangeT >

RangeT mlir::tt::ttnn::AllReduceOpGenericAdaptor< 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