public:

QuantizationParams();

QuantizationParams(const int32_t* zp, const float* scale, int num_channels);

// TODO Move to cpp file

QuantizationParams(const QuantizationParams&) = default;

QuantizationParams& operator=(const QuantizationParams&) = default;

QuantizationParams(QuantizationParams&&) = default;

QuantizationParams& operator=(QuantizationParams&&) = default;

virtual ~QuantizationParams();

virtual int32_t get_zeroP_for_channel(int i) const;

virtual float get_scale_for_channel(int i) const;

inline int num_channels() const { return _num_channels; };

// Allocate the tensor metadata on a different heap from the data scratch pads

// Note: as long as derived classes dont override new and delete, these will

// get called correctly

void* operator new(size_t sz);

void operator delete(void* p);

protected:

const int32_t* _zp;

const float* _scale;

int _num_channels;

public:

PerTensorQuantizationParams(const int32_t& zp, const float& scale);

virtual int32_t get_zeroP_for_channel(int i) const override;

virtual float get_scale_for_channel(int i) const override;

// virtual inline const int num_channels() { return _num_channels; };

public:

// PerChannelQuantizationParams(const int32_t* zp, const int32_t* scale, int

// num_channels) : _zp(zp), _scale(scale), _num_channels(num_channels) {}

// Smarter compile time thing might be

template <int num_channels>

PerChannelQuantizationParams(const int32_t (&zp)[num_channels],

const float (&scale)[num_channels])

: QuantizationParams(zp, scale, num_channels) {}

virtual int32_t get_zeroP_for_channel(int i) const override;

virtual float get_scale_for_channel(int i) const override;

public:

QuantizationParams* operator->();

const QuantizationParams* operator->() const;

// As long as operating on instantiations of this class and not pointers this

// function will work

const QuantizationParams* operator*() const;

QuantizationParamsHandle();

QuantizationParamsHandle(QuantizationParams* ptr);

QuantizationParamsHandle& operator=(QuantizationParams* ptr);

QuantizationParamsHandle(QuantizationParamsHandle&& that);

QuantizationParamsHandle& operator=(QuantizationParamsHandle&& that);

~QuantizationParamsHandle();

void free();

//// Force everything to be on the utensor allocator

//void* operator new(size_t sz);

//void operator delete(void* p);

// KEY BIT

friend class AllocatorInterface;

friend class TensorInterface;