Skip to content
Snippets Groups Projects
Commit 95ba99c9 authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

complete the PTQ float to double migration

parent 064fef3b
No related branches found
No related tags found
2 merge requests!54Update 0.3.1 -> 0.4.0,!36Global Quantization Improvements
......@@ -49,7 +49,7 @@ static void rescaleTensor(std::shared_ptr<Tensor> tensor, double scaling)
mulOp.setDataType(tensor->dataType());
mulOp.setBackend(tensor->backend());
std::shared_ptr<Aidge::Tensor> scalingTensor = std::make_shared<Aidge::Tensor>(Aidge::Array1D<float, 1> {scaling});
std::shared_ptr<Aidge::Tensor> scalingTensor = std::make_shared<Aidge::Tensor>(Aidge::Array1D<double, 1> {scaling});
scalingTensor->setDataType(tensor->dataType());
scalingTensor->setBackend(tensor->backend());
......@@ -120,12 +120,7 @@ void crossLayerEqualization(std::shared_ptr<GraphView> graphView, double targetD
if (isAffine(node))
affineNodeVector.push_back(node);
if (affineNodeVector.empty()) {
Log::notice("No affine nodes found in the network. CLE cannot be applied.");
return;
}
double maxRangeDelta;
int iteration = 0;
do
{
......
......@@ -72,13 +72,13 @@ bool checkArchitecture(std::shared_ptr<GraphView> graphView)
return true;
}
static void rescaleTensor(std::shared_ptr<Tensor> tensor, float scaling)
static void rescaleTensor(std::shared_ptr<Tensor> tensor, double scaling)
{
auto mulOp = Mul_Op();
mulOp.setDataType(tensor->dataType());
mulOp.setBackend(tensor->backend());
std::shared_ptr<Aidge::Tensor> scalingTensor = std::make_shared<Aidge::Tensor>(Aidge::Array1D<float, 1> {scaling});
std::shared_ptr<Aidge::Tensor> scalingTensor = std::make_shared<Aidge::Tensor>(Aidge::Array1D<double, 1> {scaling});
scalingTensor->setDataType(tensor->dataType());
scalingTensor->setBackend(tensor->backend());
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment