colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix FullyConnect layer crash 

Signed-off-by: xiang.zhang <xiang.zhang@verisilicon.com>
This commit is contained in:
xiang.zhang 2021-06-16 15:45:12 +08:00 committed by Sven
parent 64989c6b4a
commit 574c036a69
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tim/vx/ops/fullyconnected.cc
View File

@ -36,8 +36,8 @@ FullyConnected::FullyConnected(Graph* graph, uint32_t axis)
FullyConnected::FullyConnected(Graph* graph, uint32_t axis, uint32_t weights) FullyConnected::FullyConnected(Graph* graph, uint32_t axis, uint32_t weights)
: Operation(graph, VSI_NN_OP_FCL2) { : Operation(graph, VSI_NN_OP_FCL2) {
(void)weights;
this->impl()->node()->nn_param.fcl.axis = axis; this->impl()->node()->nn_param.fcl.axis = axis;
this->impl()->node()->nn_param.fcl.weights = weights;
} }
} // namespace ops } // namespace ops