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

Use FCL2 instead of FCL which supports axis 

Signed-off-by: Kainan Cha <kainan.zha@verisilicon.com>
This commit is contained in:
Kainan Cha 2021-03-29 17:08:22 +08:00
parent c141416238
commit c569555f1f
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/tim/vx/ops/README.md
View File

@ -3,14 +3,14 @@ TIM-VX API |Internal Op |Status
Add|ADD|Mapped Add|ADD|Mapped
Multiply|MULTIPLY|Mapped Multiply|MULTIPLY|Mapped
Conv2d|CONV2D|Mapped Conv2d|CONV2D|Mapped
||CONV_RELU|Unmapped ||CONV_RELU|Deprecated
||CONV_RELU_POOL|Unmapped ||CONV_RELU_POOL|Deprecated
FullyConnected|FCL|Mapped ||FCL|Deprecated
||FCL_RELU|Unmapped ||FCL_RELU|Deprecated
Softmax|SOFTMAX|Mapped Softmax|SOFTMAX|Mapped
Pool2d|POOL|Mapped Pool2d|POOL|Mapped
LeakyRelu|LEAKY_RELU|Mapped LeakyRelu|LEAKY_RELU|Mapped
||LRN|Unmapped ||LRN|Deprecated
Concat|CONCAT|Mapped Concat|CONCAT|Mapped
Split|SPLIT|Mapped Split|SPLIT|Mapped
||NOOP|Unmapped ||NOOP|Unmapped
@ -28,7 +28,7 @@ Relu|RELU|Mapped
||REORG|Unmapped ||REORG|Unmapped
||VARIABLE|Unmapped ||VARIABLE|Unmapped
L2Normalization|L2_NORMALIZE|Mapped L2Normalization|L2_NORMALIZE|Mapped
||FCL2|Unmapped FullyConnected|FCL2|Mapped
||POOLWITHARGMAX|Unmapped ||POOLWITHARGMAX|Unmapped
||ARGMAX|Unmapped ||ARGMAX|Unmapped
Maximum|MAXIMUM|Mapped Maximum|MAXIMUM|Mapped

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

@ -31,11 +31,11 @@ namespace vx {
namespace ops { namespace ops {
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_FCL) { : Operation(graph, VSI_NN_OP_FCL2) {
this->impl()->node()->nn_param.fcl.axis = axis; this->impl()->node()->nn_param.fcl.axis = axis;
this->impl()->node()->nn_param.fcl.weights = weights; this->impl()->node()->nn_param.fcl.weights = weights;
} }
} // namespace ops } // namespace ops
} // namespace vx } // namespace vx
} // namespace tim } // namespace tim