diff options
author | Ellen Norris-Thompson <ellen.norris-thompson@arm.com> | 2019-06-24 09:23:38 +0100 |
---|---|---|
committer | Ellen Norris-Thompson <ellen.norris-thompson@arm.com> | 2019-06-25 17:12:20 +0100 |
commit | e0dbedfa9fad91c94ae610c48057fc7c24ceb0d4 (patch) | |
tree | 438d24b8411f41d155a5eb4d3b088d87ca3e2766 /src/backends/backendsCommon/WorkloadFactory.cpp | |
parent | e440329a22e03e1813ecaaa9dad34a7d9dfd446e (diff) | |
download | armnn-e0dbedfa9fad91c94ae610c48057fc7c24ceb0d4.tar.gz |
IVGCVSW-3250 Extend the Gather workload for QSymm16 support
* Add LayerTests and EndToEndTests for QSymm16
* Updated GatherQueueDescriptor::Validate function
* Reimplemented RefLayerSupport::IsGatherSupported
Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com>
Change-Id: I5b4df55d31b5ee3c5563d2df17b412a387c27595
Diffstat (limited to 'src/backends/backendsCommon/WorkloadFactory.cpp')
-rw-r--r-- | src/backends/backendsCommon/WorkloadFactory.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp index 2fba3b7059..7cda3fe884 100644 --- a/src/backends/backendsCommon/WorkloadFactory.cpp +++ b/src/backends/backendsCommon/WorkloadFactory.cpp @@ -323,7 +323,7 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId, const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); result = layerSupportObject->IsGatherSupported(OverrideDataType(input0, dataType), - OverrideDataType(input1, dataType), + input1, OverrideDataType(output, dataType), reason); break; |