7 #include <builders/ie_layer_fragment.hpp>
17 class INFERENCE_ENGINE_API_CLASS(SoftMaxLayer):
public LayerFragment {
23 explicit SoftMaxLayer(
const std::string& name =
"");
28 explicit SoftMaxLayer(Layer& genLayer);
34 SoftMaxLayer& setName(
const std::string& name);
40 const Port& getPort()
const;
46 SoftMaxLayer& setPort(
const Port& port);
51 size_t getAxis()
const;
57 SoftMaxLayer& setAxis(
size_t axis);
Definition: ie_argmax_layer.hpp:11
a header file for the Inference Engine Network interface