The class represents a builder for ArgMax layer.
More...
#include <ie_argmax_layer.hpp>
The class represents a builder for ArgMax layer.
- Deprecated:
- Use ngraph API instead.
§ ArgMaxLayer() [1/3]
InferenceEngine::Builder::ArgMaxLayer::ArgMaxLayer |
( |
const std::string & |
name = "" | ) |
|
|
explicit |
The constructor creates a builder with the name.
- Parameters
-
§ ArgMaxLayer() [2/3]
InferenceEngine::Builder::ArgMaxLayer::ArgMaxLayer |
( |
const Layer::Ptr & |
layer | ) |
|
|
explicit |
The constructor creates a builder from generic builder.
- Parameters
-
layer | pointer to generic builder |
§ ArgMaxLayer() [3/3]
InferenceEngine::Builder::ArgMaxLayer::ArgMaxLayer |
( |
const Layer::CPtr & |
layer | ) |
|
|
explicit |
The constructor creates a builder from generic builder.
- Parameters
-
layer | constant pointer to generic builder |
§ getAxis()
int InferenceEngine::Builder::ArgMaxLayer::getAxis |
( |
| ) |
const |
Returns axis.
- Returns
- Axis
§ getOutMaxVal()
size_t InferenceEngine::Builder::ArgMaxLayer::getOutMaxVal |
( |
| ) |
const |
Returns output maximum value.
- Returns
- Output maximum value
§ getPort()
const Port& InferenceEngine::Builder::ArgMaxLayer::getPort |
( |
| ) |
const |
Returns port with shapes for the layer.
- Returns
- Port with shapes
§ getTopK()
size_t InferenceEngine::Builder::ArgMaxLayer::getTopK |
( |
| ) |
const |
Returns top K.
- Returns
- Top K
§ setAxis()
ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setAxis |
( |
int |
axis | ) |
|
Sets axis.
- Parameters
-
- Returns
- reference to layer builder
§ setName()
ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setName |
( |
const std::string & |
name | ) |
|
Sets the name for the layer.
- Parameters
-
- Returns
- reference to layer builder
§ setOutMaxVal()
ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setOutMaxVal |
( |
size_t |
size | ) |
|
Sets output maximum value.
- Parameters
-
- Returns
- reference to layer builder
§ setPort()
ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setPort |
( |
const Port & |
port | ) |
|
Sets port shapes for the layer.
- Parameters
-
- Returns
- reference to layer builder
§ setTopK()
ArgMaxLayer& InferenceEngine::Builder::ArgMaxLayer::setTopK |
( |
size_t |
topK | ) |
|
Sets top K.
- Parameters
-
- Returns
- reference to layer builder
The documentation for this class was generated from the following file: