7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
18 class INFERENCE_ENGINE_API_CLASS(ReorgYoloLayer):
public LayerDecorator {
24 explicit ReorgYoloLayer(
const std::string& name =
"");
29 explicit ReorgYoloLayer(
const Layer::Ptr& layer);
34 explicit ReorgYoloLayer(
const Layer::CPtr& layer);
40 ReorgYoloLayer& setName(
const std::string& name);
46 const Port& getInputPort()
const;
52 ReorgYoloLayer& setInputPort(
const Port& ports);
57 const Port& getOutputPort()
const;
63 ReorgYoloLayer& setOutputPort(
const Port& port);
68 int getStride()
const;
74 ReorgYoloLayer& setStride(
int stride);
Definition: ie_argmax_layer.hpp:11