diff options
Diffstat (limited to 'src/backends/gpuFsa/layers/UtilsGpuFsa.cpp')
| -rw-r--r-- | src/backends/gpuFsa/layers/UtilsGpuFsa.cpp | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/layers/UtilsGpuFsa.cpp b/src/backends/gpuFsa/layers/UtilsGpuFsa.cpp index a1d96f0ec1..b11f91cc6c 100644 --- a/src/backends/gpuFsa/layers/UtilsGpuFsa.cpp +++ b/src/backends/gpuFsa/layers/UtilsGpuFsa.cpp @@ -58,4 +58,20 @@ CreatePool2dAttributes(const Pooling2dDescriptor& descriptor) pool2dAttributes.exclude_padding(excludePadding); return pool2dAttributes; +} + +arm_compute::experimental::dynamic_fusion::ResizeAttributes +CreateResizeAttributes(const armnn::ResizeDescriptor& descriptor) +{ + arm_compute::experimental::dynamic_fusion::ResizeAttributes resizeAttributes{}; + resizeAttributes.output_width(static_cast<int32_t>(descriptor.m_TargetWidth)); + resizeAttributes.output_height(static_cast<int32_t>(descriptor.m_TargetHeight)); + resizeAttributes.interpolation_policy(descriptor.m_Method == ResizeMethod::Bilinear ? + arm_compute::InterpolationPolicy::BILINEAR : + arm_compute::InterpolationPolicy::NEAREST_NEIGHBOR); + resizeAttributes.sampling_policy(descriptor.m_HalfPixelCenters ? arm_compute::SamplingPolicy::CENTER + : arm_compute::SamplingPolicy::TOP_LEFT); + resizeAttributes.align_corners(descriptor.m_AlignCorners); + + return resizeAttributes; }
\ No newline at end of file |