COMPMID-556: Allocate trainable data tensor after configuration
Change-Id: I8a5fd2418dd104a06a0972abf088f64d7c2796d3
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/112779
Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
diff --git a/src/graph/nodes/FullyConnectedLayer.cpp b/src/graph/nodes/FullyConnectedLayer.cpp
index 41f679b..219e0f9 100644
--- a/src/graph/nodes/FullyConnectedLayer.cpp
+++ b/src/graph/nodes/FullyConnectedLayer.cpp
@@ -88,6 +88,9 @@
node_ctx.add_input(_biases.set_target(_target_hint));
node_ctx.add_output(out);
+ // Configure operation
+ auto func = OperationRegistry::get().find_operation(OperationType::FullyConnectedLayer, _target_hint)->configure(node_ctx);
+
// Fill biases
if(!weights_are_loaded)
{
@@ -99,5 +102,5 @@
}
// Get function
- return OperationRegistry::get().find_operation(OperationType::FullyConnectedLayer, _target_hint)->configure(node_ctx);
+ return func;
}