Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 337
Star 4
Fork 1
P
Paddle-Lite
未验证 提交 f9b9ad31 编写于 作者: S suiyang 提交者： GitHub
浏览文件
操作

Update op_param.h

上级 76b0f930
隐藏空白更改
内联 并排
Showing with 2 addition and 0 deletion
src/operators/op_param.h
浏览文件 @ f9b9ad31
...@@ -442,6 +442,7 @@ class ElementwiseAddParam : OpParam { ...@@ -442,6 +442,7 @@ class ElementwiseAddParam : OpParam {
#endif #endif
}; };
#ifdef ElementwiseMul_OP
template <typename Dtype> template <typename Dtype>
class ElementwiseMulParam : OpParam { class ElementwiseMulParam : OpParam {
typedef typename DtypeTensorTrait<Dtype>::gtype GType; typedef typename DtypeTensorTrait<Dtype>::gtype GType;
...@@ -480,6 +481,7 @@ class ElementwiseMulParam : OpParam { ...@@ -480,6 +481,7 @@ class ElementwiseMulParam : OpParam {
void SetFpgaArgs(const fpga::EWMulArgs &args) { fpga_EW_mul_args = args; } void SetFpgaArgs(const fpga::EWMulArgs &args) { fpga_EW_mul_args = args; }
#endif #endif
}; };
#endif
#ifdef FUSION_ELEMENTWISEADDRELU_OP #ifdef FUSION_ELEMENTWISEADDRELU_OP
template <typename Dtype> template <typename Dtype>
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或