From c569555f1fcf951ea0d829a8b492ffbf5d7c2190 Mon Sep 17 00:00:00 2001 From: Kainan Cha Date: Mon, 29 Mar 2021 17:08:22 +0800 Subject: [PATCH] Use FCL2 instead of FCL which supports axis Signed-off-by: Kainan Cha --- src/tim/vx/ops/README.md | 12 ++++++------ src/tim/vx/ops/fullyconnected.cc | 4 ++-- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/tim/vx/ops/README.md b/src/tim/vx/ops/README.md index 481f59e..9586ca2 100644 --- a/src/tim/vx/ops/README.md +++ b/src/tim/vx/ops/README.md @@ -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 diff --git a/src/tim/vx/ops/fullyconnected.cc b/src/tim/vx/ops/fullyconnected.cc index c6147a3..803fa33 100644 --- a/src/tim/vx/ops/fullyconnected.cc +++ b/src/tim/vx/ops/fullyconnected.cc @@ -31,11 +31,11 @@ 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; } } // namespace ops } // namespace vx -} // namespace tim \ No newline at end of file +} // namespace tim