diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/operators/GpuPool2d.cpp')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/operators/GpuPool2d.cpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/operators/GpuPool2d.cpp b/src/dynamic_fusion/sketch/gpu/operators/GpuPool2d.cpp index a07ad00155..c602f45164 100644 --- a/src/dynamic_fusion/sketch/gpu/operators/GpuPool2d.cpp +++ b/src/dynamic_fusion/sketch/gpu/operators/GpuPool2d.cpp @@ -60,6 +60,17 @@ bool GpuPool2dSettings::mixed_precision() const return _mixed_precision; } +GpuPool2dSettings GpuPool2dSettings::use_inf_as_limit(bool use_inf_as_limit) +{ + _use_inf_as_limit = use_inf_as_limit; + return *this; +} + +bool GpuPool2dSettings::use_inf_as_limit() const +{ + return _use_inf_as_limit; +} + Status GpuPool2d::validate_op(const GpuWorkloadSketch &sketch, const ITensorInfo *src, const ITensorInfo *dst, |