From 719122c0c1b86a2272aa0eb19e9c8da8c0db2b7c Mon Sep 17 00:00:00 2001
From: Hendrik Borras <hendrikborras@web.de>
Date: Thu, 19 Aug 2021 17:18:38 +0100
Subject: [PATCH] Updated documentation for upsampling.
---
src/finn/custom_op/fpgadataflow/upsampler.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/upsampler.py b/src/finn/custom_op/fpgadataflow/upsampler.py
index bd20a3e00..7644af37a 100644
--- a/src/finn/custom_op/fpgadataflow/upsampler.py
+++ b/src/finn/custom_op/fpgadataflow/upsampler.py
@@ -10,9 +10,10 @@ from finn.util.data_packing import npy_to_rtlsim_input, rtlsim_output_to_npy
# ToDo: Fill out functions, which currently throw NotImplemented errors.
class UpsampleNearestNeighbour_Batch(HLSCustomOp):
- """Corresponds to finn-hlslib UpsampleNearestNeighbour_Batch function.
- Upsampling with the Nearest Neighbour algorithm.
- Expects square feature maps.
+ """
+ Corresponds to finn-hlslib UpsampleNearestNeighbour_Batch function.
+ Upsampling is done with the Nearest Neighbour algorithm.
+ The layer expects square feature maps for the in and output.
"""
def __init__(self, onnx_node):
--
GitLab