Skip to content
Snippets Groups Projects

Real quantization cast for PTQ

Merged Noam Zerah requested to merge noamzerah/aidge_quantization:quantization_cast into dev
2 files
+ 2
13
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 1
10
@@ -333,16 +333,7 @@ static std::shared_ptr<Tensor> getBiasTensor(std::shared_ptr<Node> node)
std::vector<std::shared_ptr<Node>> retrieveNodeVector(std::shared_ptr<GraphView> graphView, bool newSchedule, bool verbose)
{
std::vector<std::shared_ptr<Node>> nodeVector;
SequentialScheduler scheduler(graphView);
if (newSchedule)
{
scheduler.resetScheduling();
scheduler.generateScheduling(); // old way : scheduler.forward();
}
nodeVector = scheduler.getStaticScheduling();
nodeVector = graphView->getOrderedNodes();
fixScheduling(nodeVector);
nodeVector = removeMatchingNodes(nodeVector, "Producer");
Loading