From 9aed8fb43441228343b925b42464a55042c47ca0 Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Wed, 17 Nov 2021 13:16:45 +0000 Subject: IVGCVSW-6040 Update 21.11 Doxygen Documents Signed-off-by: Nikhil Raj Change-Id: Ia36ec98c4bebc27a69103911ea3409cd7db587a5 --- ...constants_float_to_half_tests_8cpp_source.xhtml | 137 +++++++++++++++++++++ 1 file changed, 137 insertions(+) create mode 100644 21.11/_convert_constants_float_to_half_tests_8cpp_source.xhtml (limited to '21.11/_convert_constants_float_to_half_tests_8cpp_source.xhtml') diff --git a/21.11/_convert_constants_float_to_half_tests_8cpp_source.xhtml b/21.11/_convert_constants_float_to_half_tests_8cpp_source.xhtml new file mode 100644 index 0000000000..e3fb1ed61e --- /dev/null +++ b/21.11/_convert_constants_float_to_half_tests_8cpp_source.xhtml @@ -0,0 +1,137 @@ + + + + + + + + + + + + + +ArmNN: src/armnn/test/optimizations/ConvertConstantsFloatToHalfTests.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.11 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
ConvertConstantsFloatToHalfTests.cpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "../TestUtils.hpp"
7 
8 #include <Optimizer.hpp>
9 #include <Half.hpp>
10 
11 #include <doctest/doctest.h>
12 
13 using namespace armnn;
14 
15 TEST_SUITE("Optimizer")
16 {
17 using namespace armnn::optimizations;
18 
19 TEST_CASE("ConvertConstantsFloatToHalfTest")
20 {
21  armnn::Graph graph;
22 
23  const armnn::TensorInfo info({ 1, 1, 1, 2 }, armnn::DataType::Float16);
24 
25  // Create const tensor from fp32 data
26  unsigned int dims[] = { 4, 1, 1, 1 };
27  std::vector<float> floatWeights{ 1.0f, 2.0f, 3.0f, 4.0f };
28  armnn::ConstTensor weights(armnn::TensorInfo(4, dims, armnn::DataType::Float32, 0.0f, 0, true), floatWeights);
29 
30  // Create simple test network
31  auto input = graph.AddLayer<armnn::InputLayer>(0, "input");
32  input->GetOutputSlot().SetTensorInfo(info);
33 
34  auto fc = graph.AddLayer<armnn::FullyConnectedLayer>(armnn::FullyConnectedDescriptor(), "fc");
35  fc->m_Weight = std::make_unique<armnn::ScopedTensorHandle>(weights);
36  fc->GetOutputSlot().SetTensorInfo(info);
37 
38  auto output = graph.AddLayer<armnn::OutputLayer>(1, "output");
39 
40  // Connect up the layers
41  input->GetOutputSlot().Connect(fc->GetInputSlot(0));
42  fc->GetOutputSlot().Connect(output->GetInputSlot(0));
43 
44  // Check tensor data type before conversion
45  CHECK(fc->m_Weight->GetTensorInfo().GetDataType() == armnn::DataType::Float32);
46 
47  // Run the optimizer
48  armnn::Optimizer::Pass(graph, armnn::MakeOptimizations(ConvertConstantsFloatToHalf()));
49 
50  // Check tensor data type after conversion
51  CHECK(fc->m_Weight->GetTensorInfo().GetDataType() == armnn::DataType::Float16);
52 
53  // Check whether data matches expected fp16 data
54  const Half* data = fc->m_Weight->GetConstTensor<Half>();
55  CHECK(data[0] == Half(1.0f));
56  CHECK(data[1] == Half(2.0f));
57  CHECK(data[2] == Half(3.0f));
58  CHECK(data[3] == Half(4.0f));
59 }
60 
61 }
armnn::TEST_SUITE
TEST_SUITE("TestConstTensorLayerVisitor")
Definition: ConstTensorLayerVisitor.cpp:110
+
armnn::MakeOptimizations
Optimizer::Optimizations MakeOptimizations(Args &&... args)
Definition: Optimizer.hpp:43
+ + +
armnn::Graph::AddLayer
LayerT * AddLayer(Args &&... args)
Adds a new layer, of type LayerType, to the graph constructed with the arguments passed.
Definition: Graph.hpp:417
+
armnn::OutputSlot::Connect
int Connect(InputSlot &destination)
Definition: Layer.cpp:83
+
armnn::Optimizer::Pass
static void Pass(Graph &graph, const Optimizations &optimizations)
Definition: Optimizer.cpp:16
+
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
+
Optimizer.hpp
+
armnn::OutputLayer
A layer user-provided data can be bound to (e.g. inputs, outputs).
Definition: OutputLayer.hpp:13
+
armnn::optimizations::ConvertConstantsFloatToHalf
ConvertConstants< Float32ToFloat16, IsFloat16Layer > ConvertConstantsFloatToHalf
Definition: ConvertConstants.hpp:155
+
armnn::FullyConnectedLayer
This layer represents a fully connected operation.
Definition: FullyConnectedLayer.hpp:15
+
armnn::FullyConnectedLayer::m_Weight
std::shared_ptr< ConstTensorHandle > m_Weight
A unique pointer to store Weight values.
Definition: FullyConnectedLayer.hpp:20
+
armnn::FullyConnectedDescriptor
A FullyConnectedDescriptor for the FullyConnectedLayer.
Definition: Descriptors.hpp:381
+
armnn::ConstTensor
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
Definition: Tensor.hpp:327
+ + +
Half.hpp
+ +
armnn::InputLayer
A layer user-provided data can be bound to (e.g. inputs, outputs).
Definition: InputLayer.hpp:13
+
armnn::OutputSlot::SetTensorInfo
void SetTensorInfo(const TensorInfo &tensorInfo) override
Definition: Layer.cpp:58
+
armnn::Layer::GetOutputSlot
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Get the const output slot handle by slot index.
Definition: Layer.hpp:318
+ +
armnn::Half
half_float::half Half
Definition: Half.hpp:18
+
+
+ + + + -- cgit v1.2.1