ngraph::onnx_import::pooling::PoolingFactory Class Reference

Factory class which generates sub-graphs for ONNX 'regular' pooling operators. More...

#include <pooling_factory.hpp>

Inheritance diagram for ngraph::onnx_import::pooling::PoolingFactory:

Collaboration diagram for ngraph::onnx_import::pooling::PoolingFactory:

Public Member Functions
OutputVector	make_avg_pool () const
	Creates average pooling ONNX operation. More...
OutputVector	make_max_pool () const
	Creates max pooling ONNX operation. More...

Protected Member Functions
	PoolingFactory (const Node &node)

Protected Attributes
Node	m_onnx_node
const OutputVector	m_inputs
Shape	m_kernel_shape
Strides	m_strides
Strides	m_dilations
Shape	m_padding_below
Shape	m_padding_above
ngraph::op::PadType	m_auto_pad

Detailed Description

Factory class which generates sub-graphs for ONNX 'regular' pooling operators.

Note

This factory is intended for creating pooling operations like:

• AveragePool
• MaxPool

This base class holds all common attributes like srides, dilations, paddings, kernel shape and auto_pad type.

See also

GlobalPoolingFactory

Member Function Documentation

make_avg_pool()

OutputVector ngraph::onnx_import::pooling::PoolingFactory::make_avg_pool

(

)

const

Creates average pooling ONNX operation.

Returns

Vector of output nodes.

make_max_pool()

OutputVector ngraph::onnx_import::pooling::PoolingFactory::make_max_pool

(

)

const

Creates max pooling ONNX operation.

Returns

Vector of output nodes.

---

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

• frontend/onnx_import/include/onnx_import/utils/pooling_factory.hpp