aboutsummaryrefslogtreecommitdiff
path: root/utils/GraphUtils.h
diff options
context:
space:
mode:
Diffstat (limited to 'utils/GraphUtils.h')
-rw-r--r--utils/GraphUtils.h12
1 files changed, 12 insertions, 0 deletions
diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h
index 52180ca495..bc7699b70c 100644
--- a/utils/GraphUtils.h
+++ b/utils/GraphUtils.h
@@ -77,7 +77,19 @@ private:
class TFPreproccessor : public IPreprocessor
{
public:
+ /** Constructor
+ *
+ * @param[in] min_range Min normalization range. (Defaults to -1.f)
+ * @param[in] max_range Max normalization range. (Defaults to 1.f)
+ */
+ TFPreproccessor(float min_range = -1.f, float max_range = 1.f);
+
+ // Inherited overriden methods
void preprocess(ITensor &tensor) override;
+
+private:
+ float _min_range;
+ float _max_range;
};
/** PPM writer class */