7 #include <builders/ie_layer_fragment.hpp>
18 class INFERENCE_ENGINE_API_CLASS(PoolingLayer):
public LayerFragment {
40 explicit PoolingLayer(
const std::string& name =
"");
45 explicit PoolingLayer(Layer& genLayer);
50 operator Layer()
const override;
56 PoolingLayer& setName(
const std::string& name);
62 const Port& getInputPort()
const;
68 PoolingLayer& setInputPort(
const Port& port);
73 const Port& getOutputPort()
const;
79 PoolingLayer& setOutputPort(
const Port& port);
84 const std::vector<size_t> getKernel()
const;
90 PoolingLayer& setKernel(
const std::vector<size_t>& kernel);
95 const std::vector<size_t> getStrides()
const;
101 PoolingLayer& setStrides(
const std::vector<size_t>& strides);
106 const std::vector<size_t> getPaddingsBegin()
const;
112 PoolingLayer& setPaddingsBegin(
const std::vector<size_t>& paddings);
117 const std::vector<size_t> getPaddingsEnd()
const;
123 PoolingLayer& setPaddingsEnd(
const std::vector<size_t>& paddings);
128 PoolingType getPoolingType()
const;
134 PoolingLayer& setPoolingType(PoolingType type);
139 RoundingType getRoundingType()
const;
145 PoolingLayer& setRoundingType(RoundingType type);
150 bool getExcludePad()
const;
156 PoolingLayer& setExcludePad(
bool exclude);
162 static void validate(
const Layer& layer);
166 RoundingType roundingType;
Definition: ie_argmax_layer.hpp:11
a header file for the Inference Engine Network interface