7 #include <builders/ie_layer_fragment.hpp>
17 class INFERENCE_ENGINE_API_CLASS(ArgMaxLayer):
public LayerFragment {
23 explicit ArgMaxLayer(
const std::string& name =
"");
28 explicit ArgMaxLayer(Layer& genLayer);
34 ArgMaxLayer& setName(
const std::string& name);
40 const Port& getPort()
const;
46 ArgMaxLayer& setPort(
const Port& port);
57 ArgMaxLayer& setAxis(
int axis);
62 size_t getTopK()
const;
68 ArgMaxLayer& setTopK(
size_t topK);
73 size_t getOutMaxVal()
const;
79 ArgMaxLayer& setOutMaxVal(
size_t size);
Definition: ie_argmax_layer.hpp:11
a header file for the Inference Engine Network interface