A class to hold information about learnable parameters (parameter sets) and activations corresponding to a certain operation of the computational graph.
More...
#include <mlsl.hpp>
A class to hold information about learnable parameters (parameter sets) and activations corresponding to a certain operation of the computational graph.
- Returns
- The distribution used by the current operation.
size_t MLSL::Operation::GetGlobalMinibatchOffset |
( |
| ) |
|
- Returns
- The offset of the local mini-batch portion within the global mini-batch.
size_t MLSL::Operation::GetGlobalMinibatchSize |
( |
| ) |
|
- Returns
- The length of the global mini-batch.
Activation* MLSL::Operation::GetInput |
( |
size_t |
idx | ) |
|
Returns the input activation by index.
- Parameters
-
idx | the input activation's index |
- Returns
- The input activation.
size_t MLSL::Operation::GetInputCount |
( |
| ) |
|
- Returns
- The number of input activations for the current operation.
size_t MLSL::Operation::GetLocalMinibatchSize |
( |
| ) |
|
- Returns
- The length of the local mini-batch portion.
const char* MLSL::Operation::GetName |
( |
| ) |
|
- Returns
- The operation name.
OpType MLSL::Operation::GetOpType |
( |
| ) |
|
- Returns
- The operation type.
Activation* MLSL::Operation::GetOutput |
( |
size_t |
idx | ) |
|
Returns the output activation by index.
- Parameters
-
idx | the output activation's index |
- Returns
- The output activation.
size_t MLSL::Operation::GetOutputCount |
( |
| ) |
|
- Returns
- The number of output activations for the current operation.
ParameterSet* MLSL::Operation::GetParameterSet |
( |
size_t |
idx | ) |
|
Returns the parameter set by index
- Parameters
-
idx | the parameter set's index |
- Returns
- The parameter set.
size_t MLSL::Operation::GetParameterSetCount |
( |
| ) |
|
- Returns
- The number of parameter sets for the current operation.
Session* MLSL::Operation::GetSession |
( |
| ) |
|
- Returns
- The session to which the current operation belongs.
bool MLSL::Operation::HasParameterSets |
( |
| ) |
|
- Returns
- True if the current operation has parameter sets (weights or bias), false otherwise.
void MLSL::Operation::SetDistribution |
( |
Distribution * |
dist | ) |
|
Sets the operation's distribution for the case when the operation has been created without the distribution through session->AddOperation(regInfo, NULL)
void MLSL::Operation::SetNext |
( |
Operation * |
next, |
|
|
size_t |
actIdx, |
|
|
size_t |
nextOpActIdx |
|
) |
| |
Sets the next operation in the computational graph.
- Parameters
-
next | the next operation |
actIdx | the index of the current operation's output activation correspoding to the next operation |
nextOpActIdx | the index of the next operation's input activation correspoding to the current operation |
void MLSL::Operation::SetPrev |
( |
Operation * |
prev, |
|
|
size_t |
actIdx, |
|
|
size_t |
prevOpActIdx |
|
) |
| |
Sets the previous operation in the computational graph.
- Parameters
-
prev | the previous operation |
actIdx | the index of the current operation's input activation correspoding to the previous operation |
prevOpActIdx | the index of the previous operation's output activation correspoding to the current operation |
The documentation for this class was generated from the following file: