aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/RefLayerSupport.cpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2020-07-27 11:27:19 +0100
committerJim Flynn <jim.flynn@arm.com>2020-07-28 14:45:10 +0000
commit4408850bd426481e39ae9182aff4d563b6836181 (patch)
tree2fa8ad2f7ed34c249a6f9b701ad2cd0caa91a389 /src/backends/reference/RefLayerSupport.cpp
parentc135179c935e3f85e591014e14be81b3f2597825 (diff)
downloadarmnn-4408850bd426481e39ae9182aff4d563b6836181.tar.gz
IVGCVSW-4712 Add EndToEnd test for FILL
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: Ic89bcbbd580abe1b05bd26748db704e83cf65bea
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
-rw-r--r--src/backends/reference/RefLayerSupport.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 877d200208..1886d3253b 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -877,6 +877,8 @@ bool RefLayerSupport::IsFillSupported(const TensorInfo& input,
supported &= CheckSupportRule(TypeAnyOf(input, supportedTypes), reasonIfUnsupported,
"Reference Fill: input type not supported.");
+ supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported,
+ "Reference Fill: output type not supported.");
return supported;
}