diff options
Diffstat (limited to 'utils/Utils.h')
-rw-r--r-- | utils/Utils.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/utils/Utils.h b/utils/Utils.h index ec08896257..7fa74ab08b 100644 --- a/utils/Utils.h +++ b/utils/Utils.h @@ -401,7 +401,7 @@ public: void fill_tensor(T &tensor) { ARM_COMPUTE_ERROR_ON(!is_open()); - ARM_COMPUTE_ERROR_ON_DATA_TYPE_NOT_IN(&tensor, arm_compute::DataType::QASYMM8, arm_compute::DataType::S32, arm_compute::DataType::F32); + ARM_COMPUTE_ERROR_ON_DATA_TYPE_NOT_IN(&tensor, arm_compute::DataType::QASYMM8, arm_compute::DataType::S32, arm_compute::DataType::F32, arm_compute::DataType::F16); try { // Map buffer if creating a CLTensor |