diff options
Diffstat (limited to 'reference_model/src/ops/image.h')
-rw-r--r-- | reference_model/src/ops/image.h | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/reference_model/src/ops/image.h b/reference_model/src/ops/image.h index fea4885..508d2c8 100644 --- a/reference_model/src/ops/image.h +++ b/reference_model/src/ops/image.h @@ -23,7 +23,7 @@ using namespace tosa; namespace TosaReference { -template <DType InDtype, DType OutDtype> +template <DType InDtype, DType OutDtype, typename resize_t> class OpResize : public GraphNode { public: @@ -39,12 +39,9 @@ public: protected: TosaResizeAttribute* attribute; - std::vector<int32_t> output_size; - std::vector<int32_t> stride; - std::vector<int32_t> offset; - int32_t shift; - std::vector<float> stride_fp; - std::vector<float> offset_fp; + std::vector<int16_t> scale; + std::vector<int16_t> offset; + std::vector<int16_t> border; ResizeMode mode; TosaReference::TensorTemplate<TIn>* in; TosaReference::TensorTemplate<TOut>* out; |