The class represents a builder for SoftMax layer.
More...
#include <ie_softmax_layer.hpp>
The class represents a builder for SoftMax layer.
- Deprecated:
- Use ngraph API instead.
§ SoftMaxLayer() [1/3]
InferenceEngine::Builder::SoftMaxLayer::SoftMaxLayer |
( |
const std::string & |
name = "" | ) |
|
|
explicit |
The constructor creates a builder with the name.
- Parameters
-
§ SoftMaxLayer() [2/3]
InferenceEngine::Builder::SoftMaxLayer::SoftMaxLayer |
( |
const Layer::Ptr & |
layer | ) |
|
|
explicit |
The constructor creates a builder from generic builder.
- Parameters
-
layer | pointer to generic builder |
§ SoftMaxLayer() [3/3]
InferenceEngine::Builder::SoftMaxLayer::SoftMaxLayer |
( |
const Layer::CPtr & |
layer | ) |
|
|
explicit |
The constructor creates a builder from generic builder.
- Parameters
-
layer | constant pointer to generic builder |
§ getAxis()
size_t InferenceEngine::Builder::SoftMaxLayer::getAxis |
( |
| ) |
const |
Returns axis.
- Returns
- Axis
§ getPort()
const Port& InferenceEngine::Builder::SoftMaxLayer::getPort |
( |
| ) |
const |
Returns port with shapes for the layer.
- Returns
- Port with shapes
§ setAxis()
SoftMaxLayer& InferenceEngine::Builder::SoftMaxLayer::setAxis |
( |
size_t |
axis | ) |
|
Sets axis.
- Parameters
-
- Returns
- reference to layer builder
§ setName()
SoftMaxLayer& InferenceEngine::Builder::SoftMaxLayer::setName |
( |
const std::string & |
name | ) |
|
Sets the name for the layer.
- Parameters
-
- Returns
- reference to layer builder
§ setPort()
SoftMaxLayer& InferenceEngine::Builder::SoftMaxLayer::setPort |
( |
const Port & |
port | ) |
|
Sets port shapes for the layer.
- Parameters
-
- Returns
- reference to layer builder
The documentation for this class was generated from the following file: