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

Inheritance diagram for ngraph::onnx_import::pooling::LocalPoolingFactory:

Collaboration diagram for ngraph::onnx_import::pooling::LocalPoolingFactory:

Public Member Functions
	LocalPoolingFactory (const Node &node)

Public Member Functions inherited from ngraph::onnx_import::pooling::PoolingFactory
OutputVector	make_avg_pool () const
	Creates average pooling ONNX operation. More...

OutputVector	make_max_pool () const
	Creates max pooling ONNX operation. More...

Additional Inherited Members
Protected Member Functions inherited from ngraph::onnx_import::pooling::PoolingFactory
	PoolingFactory (const Node &node)

Protected Attributes inherited from ngraph::onnx_import::pooling::PoolingFactory
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 'local' pooling operators.

Note: For a 'local' pooling operation, the kernel shape attribute is required

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

Public Member Functions