19 #include <type_traits>
21 #include "ngraph/node.hpp"
22 #include "ngraph/op/avg_pool.hpp"
23 #include "ngraph/op/max_pool.hpp"
24 #include "ngraph/op/op.hpp"
25 #include "ngraph/op/util/attr_types.hpp"
26 #include "ngraph/shape.hpp"
27 #include "ngraph/strides.hpp"
28 #include "onnx_import/core/node.hpp"
69 const OutputVector m_inputs;
73 Shape m_padding_below;
74 Shape m_padding_above;
PadType
Padding Type used for Convolution and Pooling
Definition: attr_types.hpp:71
OutputVector make_max_pool() const
Creates max pooling ONNX operation.
OutputVector make_avg_pool() const
Creates average pooling ONNX operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Factory class which generates sub-graphs for ONNX 'local' pooling operators.
Definition: pooling_factory.hpp:83
Factory class which generates sub-graphs for ONNX 'regular' pooling operators.
Definition: pooling_factory.hpp:49
Factory class which generates sub-graphs for ONNX 'global' pooling operators.
Definition: pooling_factory.hpp:95