mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-08 10:00:29 +08:00
Add RKYOLOv5 RKYOLOX RKYOLOV7 (#709)
* 更正代码格式 * 更正代码格式 * 修复语法错误 * fix rk error * update * update * update * update * update * update * update Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
@@ -138,6 +138,11 @@ void FDTensor::Resize(const std::vector<int64_t>& new_shape) {
|
||||
external_data_ptr = nullptr;
|
||||
}
|
||||
|
||||
void FDTensor::SetQuantizationInfo(int32_t zero_point,std::vector<float>& scale){
|
||||
quantized_parameter_.first = zero_point;
|
||||
quantized_parameter_.second = scale;
|
||||
}
|
||||
|
||||
void FDTensor::Resize(const std::vector<int64_t>& new_shape,
|
||||
const FDDataType& data_type,
|
||||
const std::string& tensor_name,
|
||||
@@ -450,4 +455,9 @@ FDTensor& FDTensor::operator=(FDTensor&& other) {
|
||||
return *this;
|
||||
}
|
||||
|
||||
const std::pair<int32_t, std::vector<float>>
|
||||
FDTensor::GetQuantizationInfo() const{
|
||||
return quantized_parameter_;
|
||||
}
|
||||
|
||||
} // namespace fastdeploy
|
||||
|
Reference in New Issue
Block a user