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 --- ...ons_1_1_convert_fp32_network_to_bf16_impl.xhtml | 238 +++++++++++++++++++++ 1 file changed, 238 insertions(+) create mode 100644 21.11/classarmnn_1_1optimizations_1_1_convert_fp32_network_to_bf16_impl.xhtml (limited to '21.11/classarmnn_1_1optimizations_1_1_convert_fp32_network_to_bf16_impl.xhtml') diff --git a/21.11/classarmnn_1_1optimizations_1_1_convert_fp32_network_to_bf16_impl.xhtml b/21.11/classarmnn_1_1optimizations_1_1_convert_fp32_network_to_bf16_impl.xhtml new file mode 100644 index 0000000000..bf49a1e8ab --- /dev/null +++ b/21.11/classarmnn_1_1optimizations_1_1_convert_fp32_network_to_bf16_impl.xhtml @@ -0,0 +1,238 @@ + + + + + + + + + + + + + +ArmNN: ConvertFp32NetworkToBf16Impl Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.11 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+Public Member Functions | +Protected Member Functions | +List of all members
+
+
ConvertFp32NetworkToBf16Impl Class Reference
+
+
+ +

#include <ConvertFp32NetworkToBf16.hpp>

+ + + + +

+Public Member Functions

void Run (Graph &graph, Layer &layer) const
 
+ + + + + +

+Protected Member Functions

 ConvertFp32NetworkToBf16Impl ()=default
 
 ~ConvertFp32NetworkToBf16Impl ()=default
 
+

Detailed Description

+
+

Definition at line 44 of file ConvertFp32NetworkToBf16.hpp.

+

Constructor & Destructor Documentation

+ +

◆ ConvertFp32NetworkToBf16Impl()

+ +
+
+ + + + + +
+ + + + + + + +
ConvertFp32NetworkToBf16Impl ()
+ 		+protecteddefault
+
+ +

Referenced by ConvertFp32NetworkToBf16Impl::Run().

+ +
+
+ +

◆ ~ConvertFp32NetworkToBf16Impl()

+ +
+
+ + + + + +
+ + + + + + + +
~ConvertFp32NetworkToBf16Impl ()
+ 		+protecteddefault
+
+ +

Referenced by ConvertFp32NetworkToBf16Impl::Run().

+ +
+
+

Member Function Documentation

+ +

◆ Run()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
void Run (Graph & graph,
Layer & layer 
) const
+ 		+inline
+
+ +

Definition at line 48 of file ConvertFp32NetworkToBf16.hpp.

+ +

References ConvertFp32NetworkToBf16Impl::ConvertFp32NetworkToBf16Impl(), armnn::Convolution2d, armnn::Float32, armnn::FullyConnected, Layer::GetDataType(), Layer::GetType(), armnn::InsertConvertFp32ToBf16LayersBefore(), and ConvertFp32NetworkToBf16Impl::~ConvertFp32NetworkToBf16Impl().

+
49  {
50  // Only convert Float32 To BFloat16 for the Input of Convolution2d layer and FullyConnected layer.
51  // And also convert weight data type from Float32 to Bfloat16.
52  // Do not convert bias data type.
53  if (layer.GetType() == LayerType::Convolution2d)
54  {
55  if (layer.GetDataType() == DataType::Float32)
56  {
57  InsertConvertFp32ToBf16LayersBefore(graph,layer);
58  ConvertWeight<Convolution2dLayer>(&layer);
59  }
60  }
61  else if (layer.GetType() == LayerType::FullyConnected)
62  {
63  if (layer.GetDataType() == DataType::Float32)
64  {
65  InsertConvertFp32ToBf16LayersBefore(graph,layer);
66  ConvertWeight<FullyConnectedLayer>(&layer);
67  }
68  }
69  }
+ +
armnn::InsertConvertFp32ToBf16LayersBefore
std::vector< ConvertFp32ToBf16Layer * > InsertConvertFp32ToBf16LayersBefore(Graph &graph, Layer &layer, bool expectCorrectInputType)
Definition: NetworkUtils.cpp:90
+ +
+
+
+
The documentation for this class was generated from the following file: +
+
+ + + + -- cgit v1.2.1