diff --git a/include/tim/vx/ops/maxpoolgrad.h b/include/tim/vx/ops/maxpoolgrad.h index a07895d..361a38a 100644 --- a/include/tim/vx/ops/maxpoolgrad.h +++ b/include/tim/vx/ops/maxpoolgrad.h @@ -21,7 +21,7 @@ * DEALINGS IN THE SOFTWARE. * *****************************************************************************/ -#if VSI_FEAT_OP_MAXPOOLWITHARGMAX +#ifdef VSI_FEAT_OP_MAXPOOLWITHARGMAX #ifndef TIM_VX_OPS_MAXPOOLGRAD_H_ #define TIM_VX_OPS_MAXPOOLGRAD_H_ diff --git a/include/tim/vx/ops/maxpoolwithargmax2.h b/include/tim/vx/ops/maxpoolwithargmax2.h index 4a7bd98..54e6b8c 100644 --- a/include/tim/vx/ops/maxpoolwithargmax2.h +++ b/include/tim/vx/ops/maxpoolwithargmax2.h @@ -21,7 +21,7 @@ * DEALINGS IN THE SOFTWARE. * *****************************************************************************/ -#if VSI_FEAT_OP_MAXPOOLWITHARGMAX +#ifdef VSI_FEAT_OP_MAXPOOLWITHARGMAX #ifndef TIM_VX_OPS_MAXPOOLWITHARGMAX2_H_ #define TIM_VX_OPS_MAXPOOLWITHARGMAX2_H_ diff --git a/src/tim/vx/ops/maxpoolgrad.cc b/src/tim/vx/ops/maxpoolgrad.cc index c514964..dece28e 100644 --- a/src/tim/vx/ops/maxpoolgrad.cc +++ b/src/tim/vx/ops/maxpoolgrad.cc @@ -21,7 +21,7 @@ * DEALINGS IN THE SOFTWARE. * *****************************************************************************/ -#if VSI_FEAT_OP_MAXPOOLWITHARGMAX +#ifdef VSI_FEAT_OP_MAXPOOLWITHARGMAX #include "tim/vx/ops.h" #include "vsi_nn_pub.h" #include "op_impl.h" diff --git a/src/tim/vx/ops/maxpoolgrad_test.cc b/src/tim/vx/ops/maxpoolgrad_test.cc index dddecac..ccbe81d 100644 --- a/src/tim/vx/ops/maxpoolgrad_test.cc +++ b/src/tim/vx/ops/maxpoolgrad_test.cc @@ -21,7 +21,7 @@ * DEALINGS IN THE SOFTWARE. * *****************************************************************************/ -#if VSI_FEAT_OP_MAXPOOLWITHARGMAX +#ifdef VSI_FEAT_OP_MAXPOOLWITHARGMAX #include "tim/vx/context.h" #include "tim/vx/graph.h" #include "tim/vx/ops/maxpoolgrad.h" diff --git a/src/tim/vx/ops/maxpoolwithargmax2.cc b/src/tim/vx/ops/maxpoolwithargmax2.cc index a1bdc14..b867f5d 100644 --- a/src/tim/vx/ops/maxpoolwithargmax2.cc +++ b/src/tim/vx/ops/maxpoolwithargmax2.cc @@ -21,7 +21,7 @@ * DEALINGS IN THE SOFTWARE. * *****************************************************************************/ -#if VSI_FEAT_OP_MAXPOOLWITHARGMAX +#ifdef VSI_FEAT_OP_MAXPOOLWITHARGMAX #include "tim/vx/ops/maxpoolwithargmax2.h" #include "direct_map_op_impl.h" diff --git a/src/tim/vx/ops/maxpoolwithargmax2_test.cc b/src/tim/vx/ops/maxpoolwithargmax2_test.cc index b7ddb2d..a6a1531 100644 --- a/src/tim/vx/ops/maxpoolwithargmax2_test.cc +++ b/src/tim/vx/ops/maxpoolwithargmax2_test.cc @@ -21,7 +21,7 @@ * DEALINGS IN THE SOFTWARE. * *****************************************************************************/ -#if VSI_FEAT_OP_MAXPOOLWITHARGMAX +#ifdef VSI_FEAT_OP_MAXPOOLWITHARGMAX #include "tim/vx/context.h" #include "tim/vx/graph.h" #include "tim/vx/ops/maxpoolwithargmax2.h"