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

Fix Mutliply API 

Internal ops require a scale parameter to be initialized
to 1.0f.

Signed-off-by: Kainan Cha <kainan.cha@verisilicon.com>
This commit is contained in:
Kainan Cha 2021-03-17 17:56:11 +08:00
parent f79aac314c
commit b6f0ffaef6
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/tim/vx/ops/elementwise.cc
View File

@ -54,11 +54,14 @@ DEFINE_ELEMENTWISE_BINARY_OP(Maximum, VSI_NN_OP_MAXIMUM);
DEFINE_ELEMENTWISE_BINARY_OP(Add, VSI_NN_OP_ADD); DEFINE_ELEMENTWISE_BINARY_OP(Add, VSI_NN_OP_ADD);
DEFINE_ELEMENTWISE_BINARY_OP(Sub, VSI_NN_OP_SUBTRACT); DEFINE_ELEMENTWISE_BINARY_OP(Sub, VSI_NN_OP_SUBTRACT);
DEFINE_ELEMENTWISE_BINARY_OP(Div, VSI_NN_OP_DIVIDE); DEFINE_ELEMENTWISE_BINARY_OP(Div, VSI_NN_OP_DIVIDE);
DEFINE_ELEMENTWISE_BINARY_OP(Multiply, VSI_NN_OP_MULTIPLY);
DEFINE_ELEMENTWISE_BINARY_OP(Pow, VSI_NN_OP_POW); DEFINE_ELEMENTWISE_BINARY_OP(Pow, VSI_NN_OP_POW);
#undef DEFINE_ELEMENTWISE_BINARY_OP #undef DEFINE_ELEMENTWISE_BINARY_OP
Multiply::Multiply(Graph* graph) : Operation(graph, VSI_NN_OP_MULTIPLY, 2, 1) {
this->impl()->node()->nn_param.multiply.scale = 1.0f;
};
} // namespace ops } // namespace ops
} // namespace vx } // namespace vx
} // namespace tim } // namespace tim