aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2020-07-29 09:36:41 +0100
committerKeithARM <keith.davis@arm.com>2020-07-31 11:56:31 +0000
commit4b10fef654c5d54b73cc6f09327a88976d8c0fc4 (patch)
tree4a504eacb860bf1ef3feed4b5acbb79225e480fd /src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp
parente8ac13308b84b2ee9bc930ac8d27e657300e462d (diff)
downloadarmnn-4b10fef654c5d54b73cc6f09327a88976d8c0fc4.tar.gz
IVGCVSW-4712 Fill layer datatype adjustments
* Input layer to be int32 instead of same type as output * Enable float16 end to end tests * Neon and Cl layer support check for backend Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I6bc889077c8da63eeff66bd45730ce5d8783c419
Diffstat (limited to 'src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp')
-rw-r--r--src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp14
1 files changed, 9 insertions, 5 deletions
diff --git a/src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp b/src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp
index f734a948ab..5d7601b14b 100644
--- a/src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp
+++ b/src/backends/backendsCommon/test/FillEndToEndTestImpl.hpp
@@ -39,27 +39,31 @@ void FillEndToEnd(const std::vector<armnn::BackendId>& backends)
FillDescriptor descriptor;
descriptor.m_Value = 9;
- std::vector<T> inputData{
+ std::vector<int32_t> inputData {
1, 1, 5, 3
};
- std::vector<T> expectedOutputData{
+ std::vector<float> floatExpectedOutputData {
9, 9, 9, 9, 9,
9, 9, 9, 9, 9,
9, 9, 9, 9, 9
};
+ std::vector<T> expectedOutputData = armnnUtils::QuantizedVector<T>(floatExpectedOutputData);
- TensorInfo inputInfo ({ 4 }, ArmnnType);
+ TensorInfo inputInfo ({ 4 }, DataType::Signed32);
TensorInfo outputInfo({ 1, 1, 5, 3 }, ArmnnType);
armnn::INetworkPtr network = CreateFillNetwork(inputInfo, outputInfo, descriptor);
BOOST_TEST_CHECKPOINT("create a network");
- std::map<int, std::vector<T>> inputTensorData = {{ 0, inputData }};
+ std::map<int, std::vector<int32_t>> inputTensorData = {{ 0, inputData }};
std::map<int, std::vector<T>> expectedOutputTensorData = {{ 0, expectedOutputData }};
- EndToEndLayerTestImpl<ArmnnType, ArmnnType>(move(network), inputTensorData, expectedOutputTensorData, backends);
+ EndToEndLayerTestImpl<DataType::Signed32, ArmnnType>(move(network),
+ inputTensorData,
+ expectedOutputTensorData,
+ backends);
}
} // anonymous namespace \ No newline at end of file