diff --git a/src/framework/load_ops.h b/src/framework/load_ops.h index a2a6da34849641b4f99310621445cb312c7d5227..dd46776219565243f23955e279f57f76f94df24a 100644 --- a/src/framework/load_ops.h +++ b/src/framework/load_ops.h @@ -206,5 +206,8 @@ LOAD_OP2(slice, CPU, MALI_GPU); LOAD_OP2(fusion_conv_bn, CPU, FPGA); LOAD_FUSION_MATCHER(fusion_conv_bn); #endif +#ifdef ELEMENTWISESUB_OP +LOAD_OP1(elementwise_sub, CPU) +#endif LOAD_OP1(quantize, CPU); LOAD_OP1(dequantize, CPU); diff --git a/src/operators/op_param.h b/src/operators/op_param.h index f4b2696b4ea2753a900b83bd1530d6ba6ea9075f..d1795ae3475b5dccb1d50cfe37bde356e70468b1 100644 --- a/src/operators/op_param.h +++ b/src/operators/op_param.h @@ -488,6 +488,7 @@ template using ElementwiseAddReluParam = ElementwiseAddParam; #endif +#ifdef ELEMENTWISESUB_OP template class ElementwiseSubParam : OpParam { typedef typename DtypeTensorTrait::gtype GType; @@ -517,6 +518,7 @@ class ElementwiseSubParam : OpParam { GType *out_; int axis_; }; +#endif #ifdef MUL_OP template