From 284c2045de0811f423a44d2e665f4f45a9e702fa Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 11 Dec 2017 17:49:18 +0000 Subject: 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 Reviewed-by: Gian Marco Iodice --- src/graph/nodes/FullyConnectedLayer.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'src/graph/nodes/FullyConnectedLayer.cpp') diff --git a/src/graph/nodes/FullyConnectedLayer.cpp b/src/graph/nodes/FullyConnectedLayer.cpp index 41f679b4ec..219e0f9a93 100644 --- a/src/graph/nodes/FullyConnectedLayer.cpp +++ b/src/graph/nodes/FullyConnectedLayer.cpp @@ -88,6 +88,9 @@ std::unique_ptr FullyConnectedLayer::instantiate_node(Gr 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 @@ std::unique_ptr FullyConnectedLayer::instantiate_node(Gr } // Get function - return OperationRegistry::get().find_operation(OperationType::FullyConnectedLayer, _target_hint)->configure(node_ctx); + return func; } -- cgit v1.2.1