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