TT-MLIR
Public Member Functions | Private Types | Private Attributes | List of all members
mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT > Class Template Reference
Inheritance diagram for mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >:
mlir::tt::ttnn::detail::SoftmaxOpGenericAdaptorBase mlir::tt::ttnn::SoftmaxOpAdaptor

Public Member Functions

 SoftmaxOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 SoftmaxOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions={})
 
 SoftmaxOpGenericAdaptor (RangeT values, ::mlir::DictionaryAttr attrs=nullptr)
 
template<typename LateInst = SoftmaxOp, typename = std::enable_if_t<std::is_same_v<LateInst, SoftmaxOp>>>
 SoftmaxOpGenericAdaptor (RangeT values, LateInst op)
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index)
 
RangeT getODSOperands (unsigned index)
 
ValueT getInput ()
 
ValueT getOutput ()
 
RangeT getOperands ()
 
- Public Member Functions inherited from mlir::tt::ttnn::detail::SoftmaxOpGenericAdaptorBase
 SoftmaxOpGenericAdaptorBase (::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions={})
 
 SoftmaxOpGenericAdaptorBase (SoftmaxOp op)
 
std::pair< unsigned, unsigned > getODSOperandIndexAndLength (unsigned index, unsigned odsOperandsSize)
 
const PropertiesgetProperties ()
 
::mlir::DictionaryAttr getAttributes ()
 
::mlir::IntegerAttr getDimensionAttr ()
 
int32_t getDimension ()
 

Private Types

using ValueT = ::llvm::detail::ValueOfRange< RangeT >
 
using Base = detail::SoftmaxOpGenericAdaptorBase
 

Private Attributes

RangeT odsOperands
 

Additional Inherited Members

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

Member Typedef Documentation

◆ Base

template<typename RangeT >
using mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::Base = detail::SoftmaxOpGenericAdaptorBase
private

◆ ValueT

template<typename RangeT >
using mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::ValueT = ::llvm::detail::ValueOfRange<RangeT>
private

Constructor & Destructor Documentation

◆ SoftmaxOpGenericAdaptor() [1/4]

template<typename RangeT >
mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::SoftmaxOpGenericAdaptor ( RangeT  values,
::mlir::DictionaryAttr  attrs,
const Properties properties,
::mlir::RegionRange  regions = {} 
)
inline

◆ SoftmaxOpGenericAdaptor() [2/4]

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

◆ SoftmaxOpGenericAdaptor() [3/4]

template<typename RangeT >
mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::SoftmaxOpGenericAdaptor ( RangeT  values,
::mlir::DictionaryAttr  attrs = nullptr 
)
inline

◆ SoftmaxOpGenericAdaptor() [4/4]

template<typename RangeT >
template<typename LateInst = SoftmaxOp, typename = std::enable_if_t<std::is_same_v<LateInst, SoftmaxOp>>>
mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::SoftmaxOpGenericAdaptor ( RangeT  values,
LateInst  op 
)
inline

Member Function Documentation

◆ getInput()

template<typename RangeT >
ValueT mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::getInput ( )
inline

◆ getODSOperandIndexAndLength()

template<typename RangeT >
std::pair<unsigned, unsigned> mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::getODSOperandIndexAndLength ( unsigned  index)
inline

◆ getODSOperands()

template<typename RangeT >
RangeT mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::getODSOperands ( unsigned  index)
inline

◆ getOperands()

template<typename RangeT >
RangeT mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::getOperands ( )
inline

◆ getOutput()

template<typename RangeT >
ValueT mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::getOutput ( )
inline

Member Data Documentation

◆ odsOperands

template<typename RangeT >
RangeT mlir::tt::ttnn::SoftmaxOpGenericAdaptor< RangeT >::odsOperands
private

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