IVGCVSW-8033 Update Arm NN to use TOSA v0.60.0
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: If1504534fbd7d6b317fdb93083dcdbd8b827764b
diff --git a/src/backends/tosaReference/TosaRefBackend.cpp b/src/backends/tosaReference/TosaRefBackend.cpp
index bce3e2c..aaac07c 100644
--- a/src/backends/tosaReference/TosaRefBackend.cpp
+++ b/src/backends/tosaReference/TosaRefBackend.cpp
@@ -137,10 +137,15 @@
tensors.insert(tensors.end(), blockTensors.begin(), blockTensors.end());
}
- // Add all mappings to main block, the TOSA Reference Model requires the full graph to be in one block called main.
- auto* block = new TosaSerializationBasicBlock("main", operators, tensors, graphInputs, graphOutputs);
+ // Add all mappings to main block.
+ auto* block = new TosaSerializationBasicBlock("main", "main", operators, tensors, graphInputs, graphOutputs);
- handler.get()->GetBlocks().push_back(block);
+ std::vector<TosaSerializationBasicBlock*> blocks;
+ blocks.emplace_back(block);
+
+ // Add blocks to the main region.
+ auto* region = new TosaSerializationRegion("main", blocks);
+ handler->GetRegions().emplace_back(region);
auto compiledBlob =
std::make_unique<PreCompiledObjectPtr>(handler.release(), DeleteAsType<TosaSerializationHandler>);