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
Multiply|MULTIPLY|Mapped
Conv2d|CONV2D|Mapped
||CONV_RELU|Unmapped
||CONV_RELU_POOL|Unmapped
FullyConnected|FCL|Mapped
||FCL_RELU|Unmapped
||CONV_RELU|Deprecated
||CONV_RELU_POOL|Deprecated
||FCL|Deprecated
||FCL_RELU|Deprecated
Softmax|SOFTMAX|Mapped
Pool2d|POOL|Mapped
LeakyRelu|LEAKY_RELU|Mapped
||LRN|Unmapped
||LRN|Deprecated
Concat|CONCAT|Mapped
Split|SPLIT|Mapped
||NOOP|Unmapped
@ -28,7 +28,7 @@ Relu|RELU|Mapped
||REORG|Unmapped
||VARIABLE|Unmapped
L2Normalization|L2_NORMALIZE|Mapped
||FCL2|Unmapped
FullyConnected|FCL2|Mapped
||POOLWITHARGMAX|Unmapped
||ARGMAX|Unmapped
Maximum|MAXIMUM|Mapped

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

@ -31,7 +31,7 @@ namespace vx {
namespace ops {
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.weights = weights;
}