{
bool supported = true;
- std::array<DataType, 4> supportedTypes =
+ std::array<DataType, 5> supportedTypes =
{
DataType::Float16,
DataType::Float32,
DataType::QAsymmU8,
- DataType::QSymmS16
+ DataType::QSymmS16,
+ DataType::Signed32
};
supported &= CheckSupportRule(TypeAnyOf(input, supportedTypes), reasonIfUnsupported,
{
return std::make_unique<RefDebugQSymm16Workload>(descriptor, info);
}
+ if (IsDataType<DataType::Signed32>(info))
+ {
+ return std::make_unique<RefDebugSigned32Workload>(descriptor, info);
+ }
return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymm8Workload>(descriptor, info);
}
const std::string& layerName,
unsigned int slotIndex);
+template void Debug<int32_t>(const TensorInfo& inputInfo,
+ const int32_t* inputData,
+ LayerGuid guid,
+ const std::string& layerName,
+ unsigned int slotIndex);
+
} // namespace armnn
template class RefDebugWorkload<DataType::Float32>;
template class RefDebugWorkload<DataType::QAsymmU8>;
template class RefDebugWorkload<DataType::QSymmS16>;
+template class RefDebugWorkload<DataType::Signed32>;
} // namespace armnn
using RefDebugFloat32Workload = RefDebugWorkload<DataType::Float32>;
using RefDebugQAsymm8Workload = RefDebugWorkload<DataType::QAsymmU8>;
using RefDebugQSymm16Workload = RefDebugWorkload<DataType::QSymmS16>;
+using RefDebugSigned32Workload = RefDebugWorkload<DataType::Signed32>;
} // namespace armnn