From: Andrey Shedko <a.shedko@samsung.com>
Date: Wed, 19 Jan 2022 09:41:41 +0000 (+0300)
Subject: [Segt] Allow non-zero input and output offset
X-Git-Tag: submit/tizen/20220919.014846~30
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=47cff7686cd8a2295f48ce1b3ccf72e65e5acdf9;p=platform%2Fadaptation%2Fnpu%2Ftrix-engine.git

[Segt] Allow non-zero input and output offset

This commit allows input and output segments to have non-zero statring
offsets.

Signed-off-by: Andrey Shedko <a.shedko@samsung.com>
---

diff --git a/src/core/ne-model.cc b/src/core/ne-model.cc
index 3525d24..aa20ec0 100644
--- a/src/core/ne-model.cc
+++ b/src/core/ne-model.cc
@@ -200,11 +200,9 @@ Metadata_v3::createInstance (npubin_meta *meta) {
 
   if (metadata->getSegmentsNum () > MAX_SEGMENTS)
     goto sanity_violation;
-  if (metadata->getInputNum () > metadata->getSegmentsNum () ||
-      metadata->getInputNum () > MAX_TENSORS)
+  if (metadata->getInputNum () > MAX_TENSORS)
     goto sanity_violation;
-  if (metadata->getOutputNum () > metadata->getSegmentsNum () ||
-      metadata->getOutputNum () > MAX_TENSORS)
+  if (metadata->getOutputNum () > MAX_TENSORS)
     goto sanity_violation;
 
   if (metadata->getWeightSize () > 0 &&
diff --git a/src/core/ne-model.h b/src/core/ne-model.h
index 45f127d..09d43d2 100644
--- a/src/core/ne-model.h
+++ b/src/core/ne-model.h
@@ -233,8 +233,8 @@ class Metadata_v3 : public Metadata {
   uint32_t getInputNum () const override { return meta_->input_seg_num; }
   uint32_t getOutputNum () const override { return meta_->output_seg_num; }
 
-  uint32_t getInputOffset (uint32_t idx) const override { return 0; }
-  uint32_t getOutputOffset (uint32_t idx) const override { return 0; }
+  uint32_t getInputOffset (uint32_t idx) const override { return getInputSegmentOffset (idx); }
+  uint32_t getOutputOffset (uint32_t idx) const override { return getOutputSegmentOffset (idx); }
 
   uint32_t getInputTensorSize (uint32_t idx, data_layout layout) const override;
   uint32_t getOutputTensorSize (uint32_t idx, data_layout layout) const override;