Primitive which provides max pooling. More...

#include <layers_fwd.h>

Public Member Functions
	MaxPool2D (int kernel_size, int stride_length, dnnl::memory input, std::vector< dnnl::primitive > &net, std::vector< std::unordered_map< int, dnnl::memory >> &net_args, dnnl::engine eng)
	Construct a new Max Pool 2 D object. More...

Public Attributes
dnnl::memory	arg_src

dnnl::memory	arg_dst

dnnl::memory	arg_workspace

dnnl::pooling_v2_forward::primitive_desc *	pooling_fwd_pd

Detailed Description

Primitive which provides max pooling.

Constructor & Destructor Documentation

MaxPool2D::MaxPool2D	(	int	kernel_size,
		int	stride_length,
		dnnl::memory	input,
		std::vector< dnnl::primitive > &	net,
		std::vector< std::unordered_map< int, dnnl::memory >> &	net_args,
		dnnl::engine	eng
	)

Construct a new Max Pool 2 D object.

Parameters

kernel_size	the size of the kernel
stride_length	the length of the stride
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: