MLIR  21.0.0git
Classes | Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
mlir::quant::detail::UniformQuantizedSubChannelTypeStorage Struct Reference

#include "Dialect/Quant/IR/TypeDetail.h"

+ Inheritance diagram for mlir::quant::detail::UniformQuantizedSubChannelTypeStorage:

Classes

struct  KeyTy
 

Public Member Functions

 UniformQuantizedSubChannelTypeStorage (const KeyTy &key, DenseElementsAttr scales, DenseElementsAttr zeroPoints, ArrayRef< int32_t > quantizedDimensions, ArrayRef< int64_t > blockSizes)
 
bool operator== (const KeyTy &key) const
 
DenseElementsAttr getScales () const
 
DenseElementsAttr getZeroPoints () const
 
ArrayRef< int32_t > getQuantizedDimensions () const
 
ArrayRef< int64_t > getBlockSizes () const
 
- Public Member Functions inherited from mlir::quant::detail::QuantizedTypeStorage
 QuantizedTypeStorage (unsigned flags, Type storageType, Type expressedType, int64_t storageTypeMin, int64_t storageTypeMax)
 
- Public Member Functions inherited from mlir::TypeStorage
const AbstractTypegetAbstractType ()
 Return the abstract type descriptor for this type. More...
 

Static Public Member Functions

static UniformQuantizedSubChannelTypeStorageconstruct (TypeStorageAllocator &allocator, const KeyTy &key)
 Construction. More...
 
static unsigned hashKey (const KeyTy &key)
 

Public Attributes

DenseElementsAttr scales
 
DenseElementsAttr zeroPoints
 
ArrayRef< int32_t > quantizedDimensions
 
ArrayRef< int64_t > blockSizes
 
- Public Attributes inherited from mlir::quant::detail::QuantizedTypeStorage
unsigned flags
 Flags corresponding to the bitmapped enum QuantizationFlags::FlagValue. More...
 
Type storageType
 
Type expressedType
 
int64_t storageTypeMin
 
int64_t storageTypeMax
 

Additional Inherited Members

- Protected Member Functions inherited from mlir::TypeStorage
 TypeStorage ()
 This constructor is used by derived classes as part of the TypeUniquer. More...
 
- Protected Member Functions inherited from mlir::StorageUniquer::BaseStorage
 BaseStorage ()=default
 

Detailed Description

Definition at line 256 of file TypeDetail.h.

Constructor & Destructor Documentation

◆ UniformQuantizedSubChannelTypeStorage()

mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::UniformQuantizedSubChannelTypeStorage ( const KeyTy key,
DenseElementsAttr  scales,
DenseElementsAttr  zeroPoints,
ArrayRef< int32_t >  quantizedDimensions,
ArrayRef< int64_t >  blockSizes 
)
inline

Definition at line 330 of file TypeDetail.h.

Referenced by construct().

Member Function Documentation

◆ construct()

static UniformQuantizedSubChannelTypeStorage* mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::construct ( TypeStorageAllocator allocator,
const KeyTy key 
)
inlinestatic

◆ getBlockSizes()

ArrayRef<int64_t> mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::getBlockSizes ( ) const
inline

Definition at line 367 of file TypeDetail.h.

References blockSizes.

◆ getQuantizedDimensions()

ArrayRef<int32_t> mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::getQuantizedDimensions ( ) const
inline

Definition at line 363 of file TypeDetail.h.

References quantizedDimensions.

◆ getScales()

DenseElementsAttr mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::getScales ( ) const
inline

Definition at line 359 of file TypeDetail.h.

References scales.

◆ getZeroPoints()

DenseElementsAttr mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::getZeroPoints ( ) const
inline

Definition at line 361 of file TypeDetail.h.

References zeroPoints.

◆ hashKey()

static unsigned mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::hashKey ( const KeyTy key)
inlinestatic

◆ operator==()

bool mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::operator== ( const KeyTy key) const
inline

Member Data Documentation

◆ blockSizes

ArrayRef<int64_t> mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::blockSizes

Definition at line 372 of file TypeDetail.h.

Referenced by construct(), and getBlockSizes().

◆ quantizedDimensions

ArrayRef<int32_t> mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::quantizedDimensions

Definition at line 371 of file TypeDetail.h.

Referenced by construct(), and getQuantizedDimensions().

◆ scales

DenseElementsAttr mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::scales

Definition at line 369 of file TypeDetail.h.

Referenced by construct(), and getScales().

◆ zeroPoints

DenseElementsAttr mlir::quant::detail::UniformQuantizedSubChannelTypeStorage::zeroPoints

Definition at line 370 of file TypeDetail.h.

Referenced by construct(), and getZeroPoints().


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