diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-06 17:13:09 +0000 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-12-10 15:58:54 +0000 |
| commit | 5ba5e0938e68d4f90f5545a81066d56f022b376a (patch) | |
| tree | d828f8b3fd52e6d5b8f732a7ec41f832f0b921d8 /tests/validation/Helpers.cpp | |
| parent | 1d7cbb99d2a34abd15f3b6c2e017115736cd90cc (diff) | |
| download | ComputeLibrary-5ba5e0938e68d4f90f5545a81066d56f022b376a.tar.gz | |
COMPMID-1774: Implement CLGEMMReshapeLHSMatrixKernel to reshape the LHS matrix of GEMM/GEMMLowp
Change-Id: I8c5fd4c8bcdffda1522c83158981ed92baa045f4
Reviewed-on: https://review.mlplatform.org/364
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/Helpers.cpp')
| -rw-r--r-- | tests/validation/Helpers.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tests/validation/Helpers.cpp b/tests/validation/Helpers.cpp index eab6d5629f..11c454ea67 100644 --- a/tests/validation/Helpers.cpp +++ b/tests/validation/Helpers.cpp @@ -207,7 +207,7 @@ void transpose_matrix(const SimpleTensor<T> &in, SimpleTensor<T> &out) { for(int x = 0; x < width; ++x) { - const float val = in[x + y * width]; + const T val = in[x + y * width]; out[x * height + y] = val; } @@ -313,10 +313,16 @@ std::pair<int, int> get_quantized_bounds(const QuantizationInfo &quant_info, flo template void get_tile(const SimpleTensor<float> &in, SimpleTensor<float> &roi, const Coordinates &coord); template void get_tile(const SimpleTensor<half> &in, SimpleTensor<half> &roi, const Coordinates &coord); +template void get_tile(const SimpleTensor<int> &in, SimpleTensor<int> &roi, const Coordinates &coord); +template void get_tile(const SimpleTensor<short> &in, SimpleTensor<short> &roi, const Coordinates &coord); +template void get_tile(const SimpleTensor<char> &in, SimpleTensor<char> &roi, const Coordinates &coord); template void zeros(SimpleTensor<float> &in, const Coordinates &anchor, const TensorShape &shape); template void zeros(SimpleTensor<half> &in, const Coordinates &anchor, const TensorShape &shape); template void transpose_matrix(const SimpleTensor<float> &in, SimpleTensor<float> &out); template void transpose_matrix(const SimpleTensor<half> &in, SimpleTensor<half> &out); +template void transpose_matrix(const SimpleTensor<int> &in, SimpleTensor<int> &out); +template void transpose_matrix(const SimpleTensor<short> &in, SimpleTensor<short> &out); +template void transpose_matrix(const SimpleTensor<char> &in, SimpleTensor<char> &out); template void matrix_multiply(const SimpleTensor<float> &a, const SimpleTensor<float> &b, SimpleTensor<float> &out); template void matrix_multiply(const SimpleTensor<half> &a, const SimpleTensor<half> &b, SimpleTensor<half> &out); |