colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reload "==" operator for quantizations of two tensor (#583) 

Reload operator "==" to check two quantization same or not

Type: New Feature
Signed-off-by: Feiyue Chen <Feiyue.Chen@verisilicon.com>
This commit is contained in:
Chen Feiyue 2023-05-10 16:58:30 +08:00 committed by GitHub
parent 308a967bcf
commit b81f7979fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/tim/vx/tensor.h
View File

@ -81,6 +81,8 @@ class Quantization {
const std::int8_t& Fl() const{ return this->fl_; }
bool operator == (const Quantization& other_quant) const;
protected:
QuantType type_{QuantType::NONE};
int32_t channel_dim_{-1};

13
src/tim/vx/tensor.cc
View File

@ -444,6 +444,19 @@ int64_t TensorSpec::GetElementByteSize() const {
int64_t TensorSpec::GetByteSize() const {
return GetElementNum() * GetElementByteSize();
}
bool Quantization::operator == (const Quantization& other_quant) const {
if (type_ != tim::vx::QuantType::DYNAMIC_FIXED_POINT){
if(type_ == other_quant.type_ &&
scales_ == other_quant.scales_ &&
zero_points_ == other_quant.zero_points_ &&
channel_dim_ == other_quant.channel_dim_)
return true;
}
else if(fl_ == other_quant.fl_) return true;
return false;
}
namespace utils{
bool Float32ToDtype(std::shared_ptr<tim::vx::Tensor> tensor, std::vector<float> fval, uint8_t* tensorData){
bool retn = true;