patch/23.02/_convert_fp32_to_fp16_layer_8cpp_source.xhtml

 //
 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 #include "ConvertFp32ToFp16Layer.hpp"

 #include "LayerCloneBase.hpp"

 #include <armnn/TypesUtils.hpp>
 #include <armnn/backends/WorkloadData.hpp>
 #include <armnn/backends/WorkloadFactory.hpp>

 namespace armnn
 {

 ConvertFp32ToFp16Layer::ConvertFp32ToFp16Layer(const char* name)
  : Layer(1, 1, LayerType::ConvertFp32ToFp16, name)
 {
 }

 std::unique_ptr<IWorkload> ConvertFp32ToFp16Layer::CreateWorkload(const IWorkloadFactory& factory) const
 {
     ConvertFp32ToFp16QueueDescriptor descriptor;
     SetAdditionalInfo(descriptor);

     return factory.CreateWorkload(LayerType::ConvertFp32ToFp16, descriptor, PrepInfoAndDesc(descriptor));
 }

 ConvertFp32ToFp16Layer* ConvertFp32ToFp16Layer::Clone(Graph& graph) const
 {
     return CloneBase<ConvertFp32ToFp16Layer>(graph, GetName());
 }

 void ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs()
 {

     VerifyLayerConnections(1, CHECK_LOCATION());

     const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape();

     VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod);

     auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() });

     ARMNN_ASSERT(inferredShapes.size() == 1);

     ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "LayerName");
 }

 void ConvertFp32ToFp16Layer::ExecuteStrategy(IStrategy& strategy) const
 {
     strategy.ExecuteStrategy(this, GetParameters(), {}, GetName());
 }

 } // namespace armnn
WorkloadData.hpp

-armnn::TensorInfo::GetShape
const TensorShape & GetShape() const
Definition: Tensor.hpp:191

-armnn::Layer::InferOutputShapes
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
Infer the shape of the output(s) based on the provided input shape(s)
Definition: Layer.cpp:397

-WorkloadFactory.hpp

-armnn::IWorkloadFactory
Definition: WorkloadFactory.hpp:22

-armnn::ConvertFp32ToFp16Layer::Clone
ConvertFp32ToFp16Layer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition: ConvertFp32ToFp16Layer.cpp:29

-armnn::LayerType::ConvertFp32ToFp16

-armnn::ConvertFp32ToFp16Layer::ConvertFp32ToFp16Layer
ConvertFp32ToFp16Layer(const char *name)
Constructor to create a ConvertFp32ToFp16Layer.
Definition: ConvertFp32ToFp16Layer.cpp:16

-armnn::ConvertFp32ToFp16Layer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the ConvertFp32ToFp16 type.
Definition: ConvertFp32ToFp16Layer.cpp:21

-armnn::Layer::VerifyShapeInferenceType
void VerifyShapeInferenceType(const TensorShape &outputShape, ShapeInferenceMethod shapeInferenceMethod)
Definition: Layer.cpp:491

-armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6

-armnn::TensorShape
Definition: Tensor.hpp:20

-armnn::InputSlot::GetConnection
const IOutputSlot * GetConnection() const override
Definition: Layer.hpp:206

-armnn::Layer::ValidateAndCopyShape
void ValidateAndCopyShape(const TensorShape &outputShape, const TensorShape &inferredShape, const ShapeInferenceMethod shapeInferenceMethod, const std::string &layerName, const unsigned int outputSlotIndex=0)
Definition: Layer.cpp:422

-armnn::IStrategy
Definition: IStrategy.hpp:16

-armnn::Layer::VerifyLayerConnections
void VerifyLayerConnections(unsigned int expectedConnections, const CheckLocation &location) const
Definition: Layer.cpp:378

-armnn::Layer::GetInputSlot
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
Definition: Layer.hpp:324

-TypesUtils.hpp

-armnn::Layer::PrepInfoAndDesc
WorkloadInfo PrepInfoAndDesc(QueueDescriptor &descriptor) const
Helper function to reduce duplication in *Layer::CreateWorkload.
Definition: Layer.hpp:396

-ARMNN_ASSERT
#define ARMNN_ASSERT(COND)
Definition: Assert.hpp:14

-armnn::IStrategy::ExecuteStrategy
virtual void ExecuteStrategy(const IConnectableLayer *layer, const armnn::BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id=0)=0

-CHECK_LOCATION
#define CHECK_LOCATION()
Definition: Exceptions.hpp:203

-armnn::Graph
Definition: Graph.hpp:30

-armnn::ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of ConvertFp32ToFp16Layer.
Definition: ConvertFp32ToFp16Layer.cpp:34

-armnn::ConvertFp32ToFp16Layer
This layer converts data type Float 32 to Float 16.
Definition: ConvertFp32ToFp16Layer.hpp:13

-armnn::ConvertFp32ToFp16QueueDescriptor
Definition: WorkloadData.hpp:452

-armnn::ConvertFp32ToFp16Layer::ExecuteStrategy
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
Definition: ConvertFp32ToFp16Layer.cpp:50

-armnn::Layer::SetAdditionalInfo
void SetAdditionalInfo(QueueDescriptor &descriptor) const
Definition: Layer.cpp:274

-armnn::Layer::GetParameters
virtual const BaseDescriptor & GetParameters() const override
If the layer has a descriptor return it.
Definition: Layer.hpp:365

-ConvertFp32ToFp16Layer.hpp

-armnn::Layer::GetOutputSlot
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Get the const output slot handle by slot index.
Definition: Layer.hpp:326

-armnn::IOutputSlot::GetTensorInfo
virtual const TensorInfo & GetTensorInfo() const =0

-armnn::Layer::GetName
const char * GetName() const override
Returns the name of the layer.
Definition: Layer.hpp:319

-armnn::IWorkloadFactory::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const
Definition: WorkloadFactory.cpp:1561

-armnn::OutputSlot::GetTensorInfo
const TensorInfo & GetTensorInfo() const override
Definition: Layer.cpp:92

-armnn::Layer::m_ShapeInferenceMethod
ShapeInferenceMethod m_ShapeInferenceMethod
Definition: Layer.hpp:423

-armnn::Layer
Definition: Layer.hpp:217

-LayerCloneBase.hpp

-armnn::LayerType
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below...
Definition: Types.hpp:466

+Go to the documentation of this file. //

+ // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.

+ // SPDX-License-Identifier: MIT

+ //

+ #include "ConvertFp32ToFp16Layer.hpp"

+

+ #include "LayerCloneBase.hpp"

+

+ #include <armnn/TypesUtils.hpp>

+ #include <armnn/backends/WorkloadData.hpp>

+ #include <armnn/backends/WorkloadFactory.hpp>

+

+ namespace armnn

+ {

+

+ ConvertFp32ToFp16Layer::ConvertFp32ToFp16Layer(const char* name)

+  : Layer(1, 1, LayerType::ConvertFp32ToFp16, name)

+ {

+ }

+

+ std::unique_ptr<IWorkload> ConvertFp32ToFp16Layer::CreateWorkload(const IWorkloadFactory& factory) const

+ {

+     ConvertFp32ToFp16QueueDescriptor descriptor;

+     SetAdditionalInfo(descriptor);

+

+     return factory.CreateWorkload(LayerType::ConvertFp32ToFp16, descriptor, PrepInfoAndDesc(descriptor));

+ }

+

+ ConvertFp32ToFp16Layer* ConvertFp32ToFp16Layer::Clone(Graph& graph) const

+ {

+     return CloneBase<ConvertFp32ToFp16Layer>(graph, GetName());

+ }

+

+ void ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs()

+ {

+

+     VerifyLayerConnections(1, CHECK_LOCATION());

+

+     const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape();

+

+     VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod);

+

+     auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() });

+

+     ARMNN_ASSERT(inferredShapes.size() == 1);

+

+     ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "LayerName");

+ }

+

+ void ConvertFp32ToFp16Layer::ExecuteStrategy(IStrategy& strategy) const

+ {

+     strategy.ExecuteStrategy(this, GetParameters(), {}, GetName());

+ }

+

+ } // namespace armnn