diff options
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 23 |
1 files changed, 21 insertions, 2 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index e503d3fb7f..13995f5c6f 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -1736,8 +1736,27 @@ TEST_CASE("RefBroadcastToEndToEndFloat32") BroadcastToEndToEnd<armnn::DataType::Float32>(defaultBackends); } -TEST_CASE("RefBroadcastToEndToEndWithElementWiseBinaryFloat32") +TEST_CASE("RefBroadcastToEndToEndWithMultiplicationFloat32") { - BroadcastToEndToEndElementWiseBinary<armnn::DataType::Float32>(defaultBackends); + BroadcastToEndToEndElementWiseBinary<armnn::DataType::Float32>(defaultBackends, + BinaryOperation::Mul); +} + +TEST_CASE("RefBroadcastToEndToEndWithDivisionFloat32") +{ + BroadcastToEndToEndElementWiseBinary<armnn::DataType::Float32>(defaultBackends, + BinaryOperation::Div); +} + +TEST_CASE("RefBroadcastToEndToEndWithAdditionFloat32") +{ + BroadcastToEndToEndElementWiseBinary<armnn::DataType::Float32>(defaultBackends, + BinaryOperation::Add); +} + +TEST_CASE("RefBroadcastToEndToEndWithSubtractionFloat32") +{ + BroadcastToEndToEndElementWiseBinary<armnn::DataType::Float32>(defaultBackends, + BinaryOperation::Sub); } } |