#include <TypeInfo.h>

Public Member Functions
	TypeInfo ()=delete

	TypeInfo (DataType type)

	TypeInfo (DataType type, float scale, int32_t zero_point)

DataType	type () const

float	scale () const

const std::vector< float > &	scales () const

int32_t	zero_point () const

const std::vector< int32_t > &	zero_points () const

const ir::Sparsity *	sparsity () const

void	quantization (float scale, int32_t zero_point)

void	quantization (std::vector< float > &&scales, std::vector< int32_t > &&zero_points)

void	sparsity (std::shared_ptr< ir::Sparsity > sparsity)

void	type (const DataType type)

Detailed Description

Definition at line 50 of file TypeInfo.h.

Constructor & Destructor Documentation

◆ TypeInfo() [1/3]

onert::ir::TypeInfo::TypeInfo ( )

delete

◆ TypeInfo() [2/3]

onert::ir::TypeInfo::TypeInfo ( DataType type )

inlineexplicit

Definition at line 55 of file TypeInfo.h.

55: _type{type}, _sparsity{nullptr} {}

onert::ir::TypeInfo::type

DataType type() const

Definition TypeInfo.h:63

◆ TypeInfo() [3/3]

onert::ir::TypeInfo::TypeInfo	(	DataType	type,
		float	scale,
		int32_t	zero_point
	)

inline

Definition at line 57 of file TypeInfo.h.

                                                           : _type{type}, _sparsity{nullptr}
  {
    quantization(scale, zero_point);
  }

References quantization(), scale(), and zero_point().

Member Function Documentation

◆ quantization() [1/2]

void onert::ir::TypeInfo::quantization	(	float	scale,
		int32_t	zero_point
	)

inline

Definition at line 69 of file TypeInfo.h.

  {
    assert(requireQuantParam(_type) || scale == 0); // Quantize param type, or scale = 0
    _quant.scales = {scale};
    _quant.zero_points = {zero_point};
  }

References onert::ir::requireQuantParam(), scale(), onert::ir::Quantization::scales, zero_point(), and onert::ir::Quantization::zero_points.

Referenced by onert::loader::BaseLoader< LoaderDomain >::loadQuantization(), and TypeInfo().

◆ quantization() [2/2]

void onert::ir::TypeInfo::quantization	(	std::vector< float > &&	scales,
		std::vector< int32_t > &&	zero_points
	)

inline

Definition at line 75 of file TypeInfo.h.

  {
    assert(requireQuantParam(_type)); // Only quantize param types can call this
    assert(scales.size() != 0);       // Not allow empty scales vector when this is called
    assert(zero_points.size() == 0 ||
           zero_points.size() == scales.size()); // Symmetric or Asymmetric
    // Not allow meaningless quantization parameters
    assert(scales.size() == 1 ||
           std::all_of(scales.begin(), scales.end(), [](float s) { return s != 0; }));
 
    _quant.scales = std::move(scales);
    _quant.zero_points = std::move(zero_points);
  }