Dense Class Reference

Dense allows to create a fully connected layer forward primitive. More...

#include <layers_fwd.h>

Public Member Functions
	Dense (int fc_output_size, dnnl::memory input, std::vector< dnnl::primitive > &net, std::vector< std::unordered_map< int, dnnl::memory >> &net_args, dnnl::engine eng)
	Construct a new Dense object. More...

Public Attributes
dnnl::memory	arg_src
	Source memory handler.
dnnl::memory	arg_dst
	Destination memory handler.
dnnl::memory	arg_bias
	Bias memory handler.
dnnl::memory	arg_weights
	Weights memory handler.

Detailed Description

Dense allows to create a fully connected layer forward primitive.

Constructor & Destructor Documentation

Dense()

Dense::Dense	(	int	fc_output_size,
		dnnl::memory	input,
		std::vector< dnnl::primitive > &	net,
		std::vector< std::unordered_map< int, dnnl::memory >> &	net_args,
		dnnl::engine	eng
	)

Construct a new Dense object.

Parameters

fc_output_size	this is the number of units inside the FC layer
input	Input memory
net	This is the vector of primitives to which we will append the FC layer primitive
net_args	This is the associated map to which we will add the arguments of the primitive
eng	oneAPI engine that will host the primitive

---

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

• src/layers/include/layers_fwd.h