org.tensorflow.op.linalg

Class LoadAndRemapMatrix.Options

java.lang.Object

org.tensorflow.op.linalg.LoadAndRemapMatrix.Options

Enclosing class:

LoadAndRemapMatrix

public static class LoadAndRemapMatrix.Options
extends Object

Optional attributes for LoadAndRemapMatrix

Method Summary

Modifier and Type	Method and Description
LoadAndRemapMatrix.Options	maxRowsInMemory(Long maxRowsInMemory) Sets the maxRowsInMemory option.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

maxRowsInMemory

public LoadAndRemapMatrix.Options maxRowsInMemory(Long maxRowsInMemory)

Sets the maxRowsInMemory option.

Parameters:

maxRowsInMemory - The maximum number of rows to load from the checkpoint at once. If less than or equal to 0, the entire matrix will be loaded into memory. Setting this arg trades increased disk reads for lower memory usage.

Returns:

this Options instance.