diff options
author | Keith Davis <keith.davis@arm.com> | 2020-01-27 15:24:59 +0000 |
---|---|---|
committer | Keith Davis Arm <keith.davis@arm.com> | 2020-01-28 17:32:52 +0000 |
commit | 5204aa8fd1da75ccc052269f358178b54bc7792f (patch) | |
tree | d3c2b354855dab0d2da5845fd46b3950fcc75841 /src/backends/backendsCommon/MakeWorkloadHelper.hpp | |
parent | d305e1a203077bdbf2e3955abd252904127675a4 (diff) | |
download | armnn-5204aa8fd1da75ccc052269f358178b54bc7792f.tar.gz |
IVGCVSW-4306 Verify the tflite Yolo v3
* Added debug layer support for QSymmS8
* QSymmS8 support for workloads
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I51af92fadc0be290629dd9198beab5abef9e351f
Diffstat (limited to 'src/backends/backendsCommon/MakeWorkloadHelper.hpp')
-rw-r--r-- | src/backends/backendsCommon/MakeWorkloadHelper.hpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/backends/backendsCommon/MakeWorkloadHelper.hpp b/src/backends/backendsCommon/MakeWorkloadHelper.hpp index 75db73c32f..f876c6b781 100644 --- a/src/backends/backendsCommon/MakeWorkloadHelper.hpp +++ b/src/backends/backendsCommon/MakeWorkloadHelper.hpp @@ -41,7 +41,7 @@ struct MakeWorkloadForType<NullWorkload> // Makes a workload for one the specified types based on the data type requirements of the tensorinfo. // Specify type void as the WorkloadType for unsupported DataType/WorkloadType combos. template <typename Float16Workload, typename Float32Workload, typename Uint8Workload, typename Int32Workload, - typename BooleanWorkload, typename QueueDescriptorType, typename... Args> + typename BooleanWorkload, typename Int8Workload, typename QueueDescriptorType, typename... Args> std::unique_ptr<IWorkload> MakeWorkloadHelper(const QueueDescriptorType& descriptor, const WorkloadInfo& info, Args&&... args) @@ -58,6 +58,8 @@ std::unique_ptr<IWorkload> MakeWorkloadHelper(const QueueDescriptorType& descrip return MakeWorkloadForType<Float32Workload>::Func(descriptor, info, std::forward<Args>(args)...); case DataType::QAsymmU8: return MakeWorkloadForType<Uint8Workload>::Func(descriptor, info, std::forward<Args>(args)...); + case DataType::QSymmS8: + return MakeWorkloadForType<Int8Workload>::Func(descriptor, info, std::forward<Args>(args)...); case DataType::Signed32: return MakeWorkloadForType<Int32Workload>::Func(descriptor, info, std::forward<Args>(args)...); case DataType::Boolean: @@ -72,14 +74,14 @@ std::unique_ptr<IWorkload> MakeWorkloadHelper(const QueueDescriptorType& descrip // Makes a workload for one the specified types based on the data type requirements of the tensorinfo. // Calling this method is the equivalent of calling the five typed MakeWorkload method with <FloatWorkload, -// FloatWorkload, Uint8Workload, NullWorkload, NullWorkload>. +// FloatWorkload, Uint8Workload, NullWorkload, NullWorkload, NullWorkload>. // Specify type void as the WorkloadType for unsupported DataType/WorkloadType combos. template <typename FloatWorkload, typename Uint8Workload, typename QueueDescriptorType, typename... Args> std::unique_ptr<IWorkload> MakeWorkloadHelper(const QueueDescriptorType& descriptor, const WorkloadInfo& info, Args&&... args) { - return MakeWorkloadHelper<FloatWorkload, FloatWorkload, Uint8Workload, NullWorkload, NullWorkload>( + return MakeWorkloadHelper<FloatWorkload, FloatWorkload, Uint8Workload, NullWorkload, NullWorkload, NullWorkload>( descriptor, info, std::forward<Args>(args)...); |