15 #include <doctest/doctest.h> 29 Connect(inputLayer, rankLayer, inputTensorInfo, 0, 0);
30 Connect(rankLayer, outputLayer, outputTensorInfo, 0, 0);
35 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
36 void RankEndToEnd(
const std::vector<armnn::BackendId>& backends)
38 using namespace armnn;
40 std::vector<float> floatInputData{
45 std::vector<T> inputData = armnnUtils::QuantizedVector<T>(floatInputData);
47 std::vector<int32_t> expectedOutputData{ 4 };
49 TensorInfo inputInfo ({ 1, 1, 5, 3 }, ArmnnType, 0.0f, 0,
true);
57 std::map<int, std::vector<T>> inputTensorData = {{ 0, inputData }};
58 std::map<int, std::vector<int32_t>> expectedOutputTensorData = {{ 0, expectedOutputData }};
60 EndToEndLayerTestImpl<ArmnnType, DataType::Signed32>(move(network),
62 expectedOutputTensorData,
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
Copyright (c) 2021 ARM Limited and Contributors.
void Connect(armnn::IConnectableLayer *from, armnn::IConnectableLayer *to, const armnn::TensorInfo &tensorInfo, unsigned int fromIndex, unsigned int toIndex)
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
static INetworkPtr Create(NetworkOptions networkOptions={})