提交 aa144987 编写于 作者: S suiyang 提交者: GitHub

Update op_param.h

上级 dbefdf23
......@@ -442,6 +442,7 @@ class ElementwiseAddParam : OpParam {
#endif
};
#ifdef ElementwiseMul_OP
template <typename Dtype>
class ElementwiseMulParam : OpParam {
typedef typename DtypeTensorTrait<Dtype>::gtype GType;
......@@ -480,6 +481,7 @@ class ElementwiseMulParam : OpParam {
void SetFpgaArgs(const fpga::EWMulArgs &args) { fpga_EW_mul_args = args; }
#endif
};
#endif
#ifdef FUSION_ELEMENTWISEADDRELU_OP
template <typename Dtype>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册