A class to represent a collection of Operation objects with the same global mini-batch size.
More...
#include <mlsl.hpp>
A class to represent a collection of Operation objects with the same global mini-batch size.
Creates and adds an Operation object to the current session.
- Parameters
-
info | the operation's registration object that holds information about activations and/or parameter sets |
dist | the distribution that will be used by the new operation (optional), can be set later with the Operation::SetDistribution() method |
void MLSL::Session::Commit |
( |
| ) |
|
Finalizes creation of the collection of Operations. Must be called after all operations are added to the session. Session-wide optimizations take place there.
Creates an object containing the operation's registration information.
- Parameters
-
opType | the compute operation type |
- Returns
- The operation's registration object.
Deletes the previously created OperationRegInfo object.
- Parameters
-
info | the operation's registration object to delete |
size_t MLSL::Session::GetGlobalMinibatchSize |
( |
| ) |
|
- Returns
- The global mini-batch size.
Operation* MLSL::Session::GetOperation |
( |
size_t |
idx | ) |
|
Returns the operation by index.
- Parameters
-
- Returns
- The operation.
size_t MLSL::Session::GetOperationCount |
( |
| ) |
|
- Returns
- The number of operations for the current session.
Returns the statistics information for the session
- Returns
- A Statistics class instance.
void MLSL::Session::RemoveOperations |
( |
| ) |
|
Removes all operations from the current session.
void MLSL::Session::SetGlobalMinibatchSize |
( |
size_t |
globalMinibatchSize | ) |
|
Sets the global mini-batch size. Must be called before any compute operation creation.
- Parameters
-
globalMinibatchSize | the global mini-batch size |
The documentation for this class was generated from the following file: