diff options
author | narpra01 <narumol.prangnawarat@arm.com> | 2019-01-23 15:23:11 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-01-23 17:13:15 +0000 |
commit | db2b160bf9e7759d0157dfa57ee940290f5170e3 (patch) | |
tree | 536fa36ebc9eb8442b96b486a10cadab28d32647 /src/backends/cl | |
parent | c625f000198218fc8d03130ee5658f73b94b2683 (diff) | |
download | armnn-db2b160bf9e7759d0157dfa57ee940290f5170e3.tar.gz |
IVGCVSW-2511 Add end to end Gather layer test
* Add end to end test for Gather operator
* Add Support for int32 to Constant layer for Ref
* Add Int32Workload
* Add RefConstantWorkload as template for float, uint8, int32
* Remove unused RefBaseConstantWorkload
* Remove unused RefConstantFloat32Workload
* Remove unused RefConstantUint8Workload
* Add support check for int32 in LayerSupport functions
Change-Id: Ic970588a49ebe2aafb12be8adef52371feacaa7b
Diffstat (limited to 'src/backends/cl')
-rw-r--r-- | src/backends/cl/ClLayerSupport.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp index cb03e8b5ae..3e35f9d52d 100644 --- a/src/backends/cl/ClLayerSupport.cpp +++ b/src/backends/cl/ClLayerSupport.cpp @@ -121,6 +121,7 @@ bool IsSupportedForDataTypeCl(Optional<std::string&> reasonIfUnsupported, floatFuncPtr, floatFuncPtr, uint8FuncPtr, + &FalseFunc<>, std::forward<Params>(params)...); } @@ -265,7 +266,8 @@ bool ClLayerSupport::IsFloorSupported(const TensorInfo& input, input.GetDataType(), &FalseFuncF16<>, &TrueFunc<>, - &FalseFuncU8<>); + &FalseFuncU8<>, + &FalseFuncI32<>); } bool ClLayerSupport::IsFullyConnectedSupported(const TensorInfo& input, |