public interface ParameterServerUpdater
| Modifier and Type | Method and Description |
|---|---|
boolean |
isAsync()
Returns true if the
given updater is async
or synchronous
updates
|
boolean |
isReady()
Returns true if the updater is
ready for a new array
|
NDArrayHolder |
ndArrayHolder()
Get the ndarray holder for this
updater
|
int |
numUpdates()
Num updates passed through
the updater
|
void |
partialUpdate(org.nd4j.linalg.api.ndarray.INDArray arr,
org.nd4j.linalg.api.ndarray.INDArray result,
long idx,
int... dimensions)
Updates result
based on arr along a particular
INDArray#tensorAlongDimension(int, int...) |
int |
requiredUpdatesForPass()
Returns the number of required
updates for a new pass
|
void |
reset()
Reset internal counters
such as number of updates accumulated.
|
boolean |
shouldReplicate()
Returns true if
the updater has accumulated enough ndarrays to
replicate to the workers
|
Map<String,Number> |
status()
Returns the current status of this parameter server
updater
|
String |
toJson()
Serialize this updater as json
|
void |
update(org.nd4j.linalg.api.ndarray.INDArray arr,
org.nd4j.linalg.api.ndarray.INDArray result)
Updates result
based on arr
|
void |
update(NDArrayMessage message)
Do an update based on the ndarray message.
|
int requiredUpdatesForPass()
boolean isReady()
boolean isAsync()
NDArrayHolder ndArrayHolder()
int numUpdates()
Map<String,Number> status()
String toJson()
void reset()
boolean shouldReplicate()
void update(NDArrayMessage message)
message - void partialUpdate(org.nd4j.linalg.api.ndarray.INDArray arr, org.nd4j.linalg.api.ndarray.INDArray result, long idx, int... dimensions)
INDArray#tensorAlongDimension(int, int...)arr - the array to updateresult - the result ndarray to updateidx - the index to updatedimensions - the dimensions to updatevoid update(org.nd4j.linalg.api.ndarray.INDArray arr, org.nd4j.linalg.api.ndarray.INDArray result)
arr - the array to updateresult - the result ndarray to updateCopyright © 2021. All rights reserved.