diff options
Diffstat (limited to 'tests/CL')
-rw-r--r-- | tests/CL/CLAccessor.h | 24 |
1 files changed, 15 insertions, 9 deletions
diff --git a/tests/CL/CLAccessor.h b/tests/CL/CLAccessor.h index 2f955653c8..9e7b73f34f 100644 --- a/tests/CL/CLAccessor.h +++ b/tests/CL/CLAccessor.h @@ -52,15 +52,16 @@ public: /** Destructor that unmaps the CL memory. */ ~CLAccessor(); - TensorShape shape() const override; - size_t element_size() const override; - size_t size() const override; - Format format() const override; - DataType data_type() const override; - int num_channels() const override; - int num_elements() const override; - PaddingSize padding() const override; - int fixed_point_position() const override; + TensorShape shape() const override; + size_t element_size() const override; + size_t size() const override; + Format format() const override; + DataType data_type() const override; + int num_channels() const override; + int num_elements() const override; + PaddingSize padding() const override; + int fixed_point_position() const override; + QuantizationInfo quantization_info() const override; const void *operator()(const Coordinates &coord) const override; void *operator()(const Coordinates &coord) override; const void *data() const; @@ -126,6 +127,11 @@ inline int CLAccessor::fixed_point_position() const return _tensor.info()->fixed_point_position(); } +inline QuantizationInfo CLAccessor::quantization_info() const +{ + return _tensor.info()->quantization_info(); +} + inline const void *CLAccessor::data() const { return _tensor.buffer(); |