diff options
Diffstat (limited to 'tests/validation/Reference.h')
-rw-r--r-- | tests/validation/Reference.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tests/validation/Reference.h b/tests/validation/Reference.h index 37a072b60a..902d04d151 100644 --- a/tests/validation/Reference.h +++ b/tests/validation/Reference.h @@ -27,6 +27,7 @@ #include "RawTensor.h" #include "Types.h" +#include <map> #include <vector> namespace arm_compute @@ -251,6 +252,16 @@ public: */ static RawTensor compute_reference_fixed_point_pixel_wise_multiplication(const TensorShape &shape, DataType dt_in0, DataType dt_in1, DataType dt_out, float scale, int fixed_point_position, ConvertPolicy convert_policy, RoundingPolicy rounding_policy); + /** Compute reference Table Lookup. + * + * @param[in] shape Shape of the input and output tensors. + * @param[in] dt_inout Data type of input/output tensor. + * @param[in] lut Input lookup table. + * + * @return Computed raw tensor. + */ + template <typename T> + static RawTensor compute_reference_table_lookup(const TensorShape &shape, DataType dt_inout, std::map<T, T> &lut); /** Compute reference threshold. * * @param[in] shape Shape of the input and output tensors. |