13 #include <boost/test/unit_test.hpp> 20 template<armnn::DataType ArmnnTypeInput>
21 INetworkPtr CreateComparisonNetwork(
const std::vector<TensorShape>& inputShapes,
22 const TensorShape& outputShape,
24 const float qScale = 1.0f,
25 const int32_t qOffset = 0)
27 using namespace armnn;
32 IConnectableLayer* comparisonLayer = net->AddComparisonLayer(descriptor,
"comparison");
34 for (
unsigned int i = 0; i < inputShapes.size(); ++i)
36 TensorInfo inputTensorInfo(inputShapes[i], ArmnnTypeInput, qScale, qOffset);
37 IConnectableLayer* input = net->AddInputLayer(boost::numeric_cast<LayerBindingId>(i));
38 Connect(input, comparisonLayer, inputTensorInfo, 0, i);
43 Connect(comparisonLayer, output, outputTensorInfo, 0, 0);
50 void ComparisonSimpleEndToEnd(
const std::vector<BackendId>& backends,
52 const std::vector<uint8_t> expectedOutput)
54 using namespace armnn;
56 const std::vector<TensorShape> inputShapes{{ 2, 2, 2, 2 }, { 2, 2, 2, 2 }};
60 INetworkPtr net = CreateComparisonNetwork<ArmnnInType>(inputShapes, outputShape, operation);
62 BOOST_TEST_CHECKPOINT(
"create a network");
64 const std::vector<TInput> input0({ 1, 1, 1, 1, 5, 5, 5, 5,
65 3, 3, 3, 3, 4, 4, 4, 4 });
67 const std::vector<TInput> input1({ 1, 1, 1, 1, 3, 3, 3, 3,
68 5, 5, 5, 5, 4, 4, 4, 4 });
70 std::map<int, std::vector<TInput>> inputTensorData = {{ 0, input0 }, { 1, input1 }};
71 std::map<int, std::vector<uint8_t>> expectedOutputData = {{ 0, expectedOutput }};
73 EndToEndLayerTestImpl<ArmnnInType, DataType::Boolean>(move(net), inputTensorData, expectedOutputData, backends);
78 void ComparisonBroadcastEndToEnd(
const std::vector<BackendId>& backends,
80 const std::vector<uint8_t> expectedOutput)
82 using namespace armnn;
84 const std::vector<TensorShape> inputShapes{{ 1, 2, 2, 3 }, { 1, 1, 1, 3 }};
88 INetworkPtr net = CreateComparisonNetwork<ArmnnInType>(inputShapes, outputShape, operation);
90 BOOST_TEST_CHECKPOINT(
"create a network");
92 const std::vector<TInput> input0({ 1, 2, 3, 1, 0, 6,
93 7, 8, 9, 10, 11, 12 });
95 const std::vector<TInput> input1({ 1, 1, 3 });
97 std::map<int, std::vector<TInput>> inputTensorData = {{ 0, input0 }, { 1, input1 }};
98 std::map<int, std::vector<uint8_t>> expectedOutputData = {{ 0, expectedOutput }};
100 EndToEndLayerTestImpl<ArmnnInType, DataType::Boolean>(move(net), inputTensorData, expectedOutputData, backends);
typename ResolveTypeImpl< DT >::Type ResolveType
static INetworkPtr Create()
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
void Connect(armnn::IConnectableLayer *from, armnn::IConnectableLayer *to, const armnn::TensorInfo &tensorInfo, unsigned int fromIndex, unsigned int toIndex)
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
A ComparisonDescriptor for the ComparisonLayer.