Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 337
Star 4
Fork 1
P
Paddle-Lite
提交 7835a331 编写于 作者: Y yaokun01
浏览文件
操作

add marco control and load elementwise sub op

上级 6bc0322f
隐藏空白更改
内联 并排
Showing with 5 addition and 0 deletion
src/framework/load_ops.h
浏览文件 @ 7835a331
...@@ -206,5 +206,8 @@ LOAD_OP2(slice, CPU, MALI_GPU); ...@@ -206,5 +206,8 @@ LOAD_OP2(slice, CPU, MALI_GPU);
LOAD_OP2(fusion_conv_bn, CPU, FPGA); LOAD_OP2(fusion_conv_bn, CPU, FPGA);
LOAD_FUSION_MATCHER(fusion_conv_bn); LOAD_FUSION_MATCHER(fusion_conv_bn);
#endif #endif
#ifdef ELEMENTWISESUB_OP
LOAD_OP1(elementwise_sub, CPU)
#endif
LOAD_OP1(quantize, CPU); LOAD_OP1(quantize, CPU);
LOAD_OP1(dequantize, CPU); LOAD_OP1(dequantize, CPU);
src/operators/op_param.h
浏览文件 @ 7835a331
...@@ -488,6 +488,7 @@ template <typename Dtype> ...@@ -488,6 +488,7 @@ template <typename Dtype>
using ElementwiseAddReluParam = ElementwiseAddParam<Dtype>; using ElementwiseAddReluParam = ElementwiseAddParam<Dtype>;
#endif #endif
#ifdef ELEMENTWISESUB_OP
template <typename Dtype> template <typename Dtype>
class ElementwiseSubParam : OpParam { class ElementwiseSubParam : OpParam {
typedef typename DtypeTensorTrait<Dtype>::gtype GType; typedef typename DtypeTensorTrait<Dtype>::gtype GType;
...@@ -517,6 +518,7 @@ class ElementwiseSubParam : OpParam { ...@@ -517,6 +518,7 @@ class ElementwiseSubParam : OpParam {
GType *out_; GType *out_;
int axis_; int axis_;
}; };
#endif
#ifdef MUL_OP #ifdef MUL_OP
template <typename Dtype> template <typename Dtype>
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或