Fully connected layer info. More...

#include <Types.h>

Collaboration diagram for FullyConnectedLayerInfo:

Public Member Functions
FullyConnectedLayerInfo &	set_weights_trained_layout (DataLayout layout)
	Sets the weights trained data layout. More...

FullyConnectedLayerInfo &	set_transpose_weights (bool should_transpose_weights)
	Sets the transpose weights flag. More...

Data Fields
ActivationLayerInfo	activation_info {}
	Fused activation to apply after the matrix multiplication. More...

DataLayout	weights_trained_layout { DataLayout::NCHW }
	Layout that the weights have been trained with. More...

bool	transpose_weights { true }
	Transpose weights if true. More...

bool	are_weights_reshaped { false }

bool	retain_internal_weights { false }
	Retain internal reshaped weights. More...

bool	enable_fast_math { false }
	Enable fast math computation. More...

bool	fp_mixed_precision { false }
	Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. More...

Detailed Description

Fully connected layer info.

Definition at line 1829 of file Types.h.

Member Function Documentation

◆ set_transpose_weights()

FullyConnectedLayerInfo& set_transpose_weights ( bool should_transpose_weights )

inline

Sets the transpose weights flag.

Parameters

[in] should_transpose_weights Boolean flag indicating if weights should be transposed

Returns: Updated object

Definition at line 1859 of file Types.h.

     {
         transpose_weights = should_transpose_weights;
         return *this;
     }

◆ set_weights_trained_layout()

FullyConnectedLayerInfo& set_weights_trained_layout ( DataLayout layout )

inline

Sets the weights trained data layout.

Parameters

[in] layout Data layout that the weights were trained with

Returns: Updated object

Definition at line 1848 of file Types.h.

     {
         weights_trained_layout = layout;
         return *this;
     }

Field Documentation

◆ activation_info

ActivationLayerInfo activation_info {}

Fused activation to apply after the matrix multiplication.

Definition at line 1832 of file Types.h.

Referenced by CpuFullyConnected::configure(), CpuFullyConnected::has_opt_impl(), arm_compute::operator<<(), FullyConnectedLayerNode::set_fused_activation(), ClFullyConnected::validate(), and CpuFullyConnected::validate().

◆ are_weights_reshaped

bool are_weights_reshaped { false }

Deprecated:: Reshape the weights tensor if false.

Definition at line 1836 of file Types.h.

Referenced by ClFullyConnected::configure(), CLFullyConnectedLayer::configure(), CpuFullyConnected::configure(), NEFullyConnectedLayer::configure(), arm_compute::test::validation::DATA_TEST_CASE(), arm_compute::operator<<(), ClFullyConnected::validate(), and CpuFullyConnected::validate().

◆ enable_fast_math

bool enable_fast_math { false }

Enable fast math computation.

Definition at line 1838 of file Types.h.

Referenced by CpuFullyConnected::configure(), arm_compute::graph::backends::detail::create_fully_connected_layer(), CpuFullyConnected::has_opt_impl(), and CpuFullyConnected::validate().

◆ fp_mixed_precision

bool fp_mixed_precision { false }

Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy.

Definition at line 1840 of file Types.h.

Referenced by arm_compute::operator<<().

◆ retain_internal_weights

bool retain_internal_weights { false }

Retain internal reshaped weights.

Definition at line 1837 of file Types.h.

Referenced by ClFullyConnected::configure(), CLFullyConnectedLayer::configure(), CpuFullyConnected::configure(), NEFullyConnectedLayer::configure(), arm_compute::operator<<(), and CpuFullyConnected::validate().

◆ transpose_weights

bool transpose_weights { true }

Transpose weights if true.

Definition at line 1835 of file Types.h.

Referenced by FullyConnectedLayerNode::compute_weights_descriptor(), CLFullyConnectedLayer::configure(), ClFullyConnected::configure(), CpuFullyConnected::configure(), NEFullyConnectedLayer::configure(), arm_compute::test::validation::DATA_TEST_CASE(), arm_compute::operator<<(), ClFullyConnected::validate(), and CpuFullyConnected::validate().

◆ weights_trained_layout

DataLayout weights_trained_layout { DataLayout::NCHW }

Layout that the weights have been trained with.

Definition at line 1834 of file Types.h.

Referenced by ClFullyConnected::configure(), CpuFullyConnected::configure(), arm_compute::operator<<(), ClFullyConnected::validate(), and CpuFullyConnected::validate().

The documentation for this struct was generated from the following file:

arm_compute/core/Types.h

Public Member Functions

Data Fields