7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
18 class INFERENCE_ENGINE_API_CLASS(PSROIPoolingLayer):
public LayerDecorator {
24 explicit PSROIPoolingLayer(
const std::string& name =
"");
29 explicit PSROIPoolingLayer(
const Layer::Ptr& layer);
34 explicit PSROIPoolingLayer(
const Layer::CPtr& layer);
40 PSROIPoolingLayer& setName(
const std::string& name);
46 const std::vector<Port>& getInputPorts()
const;
52 PSROIPoolingLayer& setInputPorts(
const std::vector<Port>& ports);
57 const Port& getOutputPort()
const;
63 PSROIPoolingLayer& setOutputPort(
const Port& port);
68 float getSpatialScale()
const;
74 PSROIPoolingLayer& setSpatialScale(
float spatialScale);
79 size_t getOutputDim()
const;
85 PSROIPoolingLayer& setOutputDim(
size_t outDim);
90 size_t getGroupSize()
const;
96 PSROIPoolingLayer& setGroupSize(
size_t size);
Definition: ie_argmax_layer.hpp:11