diff options
Diffstat (limited to 'arm_compute/dynamic_fusion')
-rw-r--r-- | arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h b/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h index 16d88af570..6e1bcdbbfd 100644 --- a/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h +++ b/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h @@ -48,8 +48,15 @@ public: /* Set mixed_precision */ GpuPool2dSettings &mixed_precision(bool mixed_precision); + /* Get using -infinity as limit flag */ + bool use_inf_as_limit() const; + + /* Set using -infinity as limit flag */ + GpuPool2dSettings use_inf_as_limit(bool use_inf_as_limit); + private: bool _mixed_precision{ false }; + bool _use_inf_as_limit{ true }; }; /** Operator interface. */ |