aboutsummaryrefslogtreecommitdiff
path: root/tests/TensorLibrary.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/TensorLibrary.h')
-rw-r--r--tests/TensorLibrary.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/tests/TensorLibrary.h b/tests/TensorLibrary.h
index 6c079b6872..5b2c5b6fd5 100644
--- a/tests/TensorLibrary.h
+++ b/tests/TensorLibrary.h
@@ -469,6 +469,7 @@ void TensorLibrary::fill_tensor_uniform(T &&tensor, std::random_device::result_t
break;
}
case DataType::S16:
+ case DataType::QS16:
{
std::uniform_int_distribution<int16_t> distribution_s16(std::numeric_limits<int16_t>::lowest(), std::numeric_limits<int16_t>::max());
fill(tensor, distribution_s16, seed_offset);