Convolution Layer Weights Information class. More...

#include <Types.h>

Public Member Functions
	WeightsInfo ()
	Default constructor. More...

	WeightsInfo (bool are_reshaped, unsigned int kernel_width, unsigned int kernel_height, unsigned int num_kernels)
	Constructor. More...

bool	are_reshaped () const
	Flag which specifies if the weights tensor has been reshaped. More...

unsigned int	num_kernels () const
	Return the number of convolution kernels. More...

std::pair< unsigned int, unsigned int >	kernel_size () const
	Return the width and height of the kernel. More...

Detailed Description

Convolution Layer Weights Information class.

This class stores the necessary information to compute convolution layer when the weights are already reshaped

Definition at line 944 of file Types.h.

Constructor & Destructor Documentation

WeightsInfo ( )

inline

Default constructor.

Definition at line 948 of file Types.h.

         : _are_reshaped(false), _kernel_width(0), _kernel_height(0), _num_kernels(0)
     {
     }

inline

Constructor.

Parameters

[in]	are_reshaped	True if the weights have been reshaped
[in]	kernel_width	Kernel width.
[in]	kernel_height	Kernel height.
[in]	num_kernels	Number of convolution kernels.

Definition at line 959 of file Types.h.

         : _are_reshaped(are_reshaped), _kernel_width(kernel_width), _kernel_height(kernel_height), _num_kernels(num_kernels)
     {
     }

bool are_reshaped ( ) const

inline

Flag which specifies if the weights tensor has been reshaped.

Definition at line 967 of file Types.h.

     {
         return _are_reshaped;
     };

std::pair<unsigned int, unsigned int> kernel_size ( ) const

inline

Return the width and height of the kernel.

Definition at line 983 of file Types.h.

     {
         return std::make_pair(_kernel_width, _kernel_height);
     }

unsigned int num_kernels ( ) const

inline

Return the number of convolution kernels.

Definition at line 975 of file Types.h.

     {
         return _num_kernels;
     };

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