7 #include <builders/ie_layer_fragment.hpp>
18 class INFERENCE_ENGINE_API_CLASS(SplitLayer):
public LayerFragment {
24 explicit SplitLayer(
const std::string& name =
"");
29 explicit SplitLayer(Layer& genLayer);
35 SplitLayer& setName(
const std::string& name);
41 const std::vector<Port>& getOutputPorts()
const;
47 SplitLayer& setOutputPorts(
const std::vector<Port>& ports);
52 const Port& getInputPort()
const;
58 SplitLayer& setInputPort(
const Port& port);
63 size_t getAxis()
const;
69 SplitLayer& setAxis(
size_t axis);
Definition: ie_argmax_layer.hpp:11
a header file for the Inference Engine Network interface