7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
18 IE_SUPPRESS_DEPRECATED_START
19 class INFERENCE_ENGINE_NN_BUILDER_API_CLASS(ArgMaxLayer):
public LayerDecorator {
25 explicit ArgMaxLayer(
const std::string& name =
"");
30 explicit ArgMaxLayer(
const Layer::Ptr& layer);
35 explicit ArgMaxLayer(
const Layer::CPtr& layer);
41 ArgMaxLayer& setName(
const std::string& name);
47 const Port& getPort()
const;
53 ArgMaxLayer& setPort(
const Port& port);
64 ArgMaxLayer& setAxis(
int axis);
69 size_t getTopK()
const;
75 ArgMaxLayer& setTopK(
size_t topK);
80 size_t getOutMaxVal()
const;
86 ArgMaxLayer& setOutMaxVal(
size_t size);
88 IE_SUPPRESS_DEPRECATED_END
Inference Engine API.
Definition: ie_argmax_layer.hpp:11