ArmNN
 20.02
Multiplication.cpp File Reference
#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>
#include <iostream>

Go to the source code of this file.

Functions

 BOOST_FIXTURE_TEST_CASE (ParseMultiplication, SimpleMultiplicationFixture)
 
 BOOST_FIXTURE_TEST_CASE (ParseMultiplicationBroadcast4D1D, MultiplicationBroadcastFixture4D1D)
 
 BOOST_FIXTURE_TEST_CASE (ParseMultiplicationBroadcast1D4D, MultiplicationBroadcastFixture1D4D)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/3]

BOOST_FIXTURE_TEST_CASE ( ParseMultiplication  ,
SimpleMultiplicationFixture   
)

Definition at line 95 of file Multiplication.cpp.

96 {
97  using armnn::DataType;
98  RunTest<4, DataType::Float32>(0, {{ "inputTensor1", { 0.0f, 1.0f, 2.0f,
99  3.0f, 4.0f, 5.0f,
100  6.0f, 7.0f, 8.0f,
101  9.0f, 10.0f, 11.0f } },
102  { "inputTensor2", { 1.0f, 1.0f, 1.0f,
103  5.0f, 5.0f, 5.0f,
104  1.0f, 1.0f, 1.0f,
105  5.0f, 5.0f, 5.0f} } },
106  {{ "outputTensor", { 0.0f, 1.0f, 2.0f,
107  15.0f, 20.0f, 25.0f,
108  6.0f, 7.0f, 8.0f,
109  45.0f, 50.0f, 55.0f } } });
110 }
DataType
Definition: Types.hpp:32

◆ BOOST_FIXTURE_TEST_CASE() [2/3]

BOOST_FIXTURE_TEST_CASE ( ParseMultiplicationBroadcast4D1D  ,
MultiplicationBroadcastFixture4D1D   
)

Definition at line 117 of file Multiplication.cpp.

118 {
119  using armnn::DataType;
120  RunTest<4, DataType::Float32>(0, {{ "inputTensor1", { 0.0f, 1.0f, 2.0f,
121  3.0f, 4.0f, 5.0f,
122  6.0f, 7.0f, 8.0f,
123  9.0f, 10.0f, 11.0f } },
124  { "inputTensor2", { 5.0f } } },
125  {{ "outputTensor", { 0.0f, 5.0f, 10.0f,
126  15.0f, 20.0f, 25.0f,
127  30.0f, 35.0f, 40.0f,
128  45.0f, 50.0f, 55.0f } } });
129 }
DataType
Definition: Types.hpp:32

◆ BOOST_FIXTURE_TEST_CASE() [3/3]

BOOST_FIXTURE_TEST_CASE ( ParseMultiplicationBroadcast1D4D  ,
MultiplicationBroadcastFixture1D4D   
)

Definition at line 136 of file Multiplication.cpp.

References BOOST_AUTO_TEST_SUITE_END().

137 {
138  using armnn::DataType;
139  RunTest<4, DataType::Float32>(0, {{ "inputTensor1", { 3.0f } },
140  { "inputTensor2", { 0.0f, 1.0f, 2.0f,
141  3.0f, 4.0f, 5.0f,
142  6.0f, 7.0f, 8.0f,
143  9.0f, 10.0f, 11.0f } } },
144  {{ "outputTensor", { 0.0f, 3.0f, 6.0f,
145  9.0f, 12.0f, 15.0f,
146  18.0f, 21.0f, 24.0f,
147  27.0f, 30.0f, 33.0f } } });
148 }
DataType
Definition: Types.hpp:32