From 574c036a694b17f31d47e63e5f5f0a7de5949966 Mon Sep 17 00:00:00 2001 From: "xiang.zhang" Date: Wed, 16 Jun 2021 15:45:12 +0800 Subject: [PATCH] Fix FullyConnect layer crash Signed-off-by: xiang.zhang --- src/tim/vx/ops/fullyconnected.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tim/vx/ops/fullyconnected.cc b/src/tim/vx/ops/fullyconnected.cc index b561d15..16088b9 100644 --- a/src/tim/vx/ops/fullyconnected.cc +++ b/src/tim/vx/ops/fullyconnected.cc @@ -36,8 +36,8 @@ FullyConnected::FullyConnected(Graph* graph, uint32_t axis) FullyConnected::FullyConnected(Graph* graph, uint32_t axis, uint32_t weights) : Operation(graph, VSI_NN_OP_FCL2) { - (void)weights; this->impl()->node()->nn_param.fcl.axis = axis; + this->impl()->node()->nn_param.fcl.weights = weights; } } // namespace ops