diff options
Diffstat (limited to 'src/backends/aclCommon/ArmComputeTensorUtils.cpp')
| -rw-r--r-- | src/backends/aclCommon/ArmComputeTensorUtils.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/aclCommon/ArmComputeTensorUtils.cpp b/src/backends/aclCommon/ArmComputeTensorUtils.cpp index 7fbaa303db..f9335058c2 100644 --- a/src/backends/aclCommon/ArmComputeTensorUtils.cpp +++ b/src/backends/aclCommon/ArmComputeTensorUtils.cpp @@ -267,6 +267,8 @@ arm_compute::PixelValue GetPixelValue(arm_compute::ITensor& input, float pixelVa case arm_compute::DataType::QASYMM8_SIGNED: case arm_compute::DataType::QSYMM8_PER_CHANNEL: return arm_compute::PixelValue(static_cast<int8_t>(pixelValue)); + case arm_compute::DataType::S32: + return arm_compute::PixelValue(static_cast<int32_t>(pixelValue)); default: throw InvalidArgumentException("Unsupported DataType: [" + std::to_string(static_cast<int>(input.info()->data_type())) + "]"); |