COMPMID-661: directconv-uint8 (#20)
Change-Id: I84f7a1ce3658be0d3c91e65096467258af48f0b6
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/94341
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/tests/Utils.h b/tests/Utils.h
index 465cba8..70def45 100644
--- a/tests/Utils.h
+++ b/tests/Utils.h
@@ -230,6 +230,7 @@
switch(data_type)
{
case DataType::U8:
+ case DataType::QASYMM8:
*reinterpret_cast<uint8_t *>(ptr) = value;
break;
case DataType::S8:
@@ -385,14 +386,18 @@
* @param[in] data_type Data type.
* @param[in] num_channels (Optional) Number of channels.
* @param[in] fixed_point_position (Optional) Number of fractional bits.
+ * @param[in] quantization_info (Optional) Quantization info for asymmetric quantized types.
*
* @return Initialized tensor of given type.
*/
template <typename T>
-inline T create_tensor(const TensorShape &shape, DataType data_type, int num_channels = 1, int fixed_point_position = 0)
+inline T create_tensor(const TensorShape &shape, DataType data_type, int num_channels = 1,
+ int fixed_point_position = 0, QuantizationInfo quantization_info = QuantizationInfo())
{
- T tensor;
- tensor.allocator()->init(TensorInfo(shape, num_channels, data_type, fixed_point_position));
+ T tensor;
+ TensorInfo info(shape, num_channels, data_type, fixed_point_position);
+ info.set_quantization_info(quantization_info);
+ tensor.allocator()->init(info);
return tensor;
}