Uses of Class
org.deeplearning4j.nn.conf.CacheMode
-
-
Uses of CacheMode in org.deeplearning4j.nn.api
Methods in org.deeplearning4j.nn.api with parameters of type CacheMode Modifier and Type Method Description voidLayer. setCacheMode(CacheMode mode)This method sets given CacheMode for current layer -
Uses of CacheMode in org.deeplearning4j.nn.conf
Fields in org.deeplearning4j.nn.conf declared as CacheMode Modifier and Type Field Description protected CacheModeComputationGraphConfiguration. cacheModeprotected CacheModeMultiLayerConfiguration.Builder. cacheModeprotected CacheModeMultiLayerConfiguration. cacheModeprotected CacheModeNeuralNetConfiguration.Builder. cacheModeprotected CacheModeNeuralNetConfiguration. cacheModeMethods in org.deeplearning4j.nn.conf that return CacheMode Modifier and Type Method Description static CacheModeCacheMode. valueOf(String name)Returns the enum constant of this type with the specified name.static CacheMode[]CacheMode. values()Returns an array containing the constants of this enum type, in the order they are declared.Methods in org.deeplearning4j.nn.conf with parameters of type CacheMode Modifier and Type Method Description MultiLayerConfiguration.BuilderMultiLayerConfiguration.Builder. cacheMode(@NonNull CacheMode cacheMode)This method defines how/if preOutput cache is handled: NONE: cache disabled (default value) HOST: Host memory will be used DEVICE: GPU memory will be used (on CPU backends effect will be the same as for HOST)NeuralNetConfiguration.BuilderNeuralNetConfiguration.Builder. cacheMode(@NonNull CacheMode cacheMode)This method defines how/if preOutput cache is handled: NONE: cache disabled (default value) HOST: Host memory will be used DEVICE: GPU memory will be used (on CPU backends effect will be the same as for HOST)NeuralNetConfiguration.ListBuilderNeuralNetConfiguration.ListBuilder. cacheMode(@NonNull CacheMode cacheMode) -
Uses of CacheMode in org.deeplearning4j.nn.conf.memory
Fields in org.deeplearning4j.nn.conf.memory with type parameters of type CacheMode Modifier and Type Field Description static Map<CacheMode,Long>MemoryReport. CACHE_MODE_ALL_ZEROSA simple Map containing all zeros for each CacheMode keyMethods in org.deeplearning4j.nn.conf.memory that return types with arguments of type CacheMode Modifier and Type Method Description static Map<CacheMode,Long>MemoryReport. cacheModeMapFor(long value)Get a map of CacheMode with all keys associated with the specified valueMethods in org.deeplearning4j.nn.conf.memory with parameters of type CacheMode Modifier and Type Method Description longLayerMemoryReport. getMemoryBytes(MemoryType memoryType, int minibatchSize, MemoryUseMode memoryUseMode, CacheMode cacheMode, DataType dataType)longMemoryReport. getMemoryBytes(MemoryType memoryType, int minibatchSize, MemoryUseMode memoryUseMode, CacheMode cacheMode)Get the memory estimate (in bytes) for the specified type of memory, using the current ND4J data typeabstract longMemoryReport. getMemoryBytes(MemoryType memoryType, int minibatchSize, MemoryUseMode memoryUseMode, CacheMode cacheMode, DataType dataType)Get the memory estimate (in bytes) for the specified type of memorylongNetworkMemoryReport. getMemoryBytes(MemoryType memoryType, int minibatchSize, MemoryUseMode memoryUseMode, CacheMode cacheMode, DataType dataType)longLayerMemoryReport. getTotalMemoryBytes(int minibatchSize, @NonNull MemoryUseMode memoryUseMode, @NonNull CacheMode cacheMode, @NonNull DataType dataType)longMemoryReport. getTotalMemoryBytes(int minibatchSize, @NonNull MemoryUseMode memoryUseMode, @NonNull CacheMode cacheMode)Get the total memory use in bytes for the given configuration (using the current ND4J data type)abstract longMemoryReport. getTotalMemoryBytes(int minibatchSize, @NonNull MemoryUseMode memoryUseMode, @NonNull CacheMode cacheMode, @NonNull DataType dataType)Get the total memory use in bytes for the given configurationlongNetworkMemoryReport. getTotalMemoryBytes(int minibatchSize, @NonNull MemoryUseMode memoryUseMode, @NonNull CacheMode cacheMode, @NonNull DataType dataType)Method parameters in org.deeplearning4j.nn.conf.memory with type arguments of type CacheMode Modifier and Type Method Description LayerMemoryReport.BuilderLayerMemoryReport.Builder. cacheMemory(Map<CacheMode,Long> cacheModeMemoryFixed, Map<CacheMode,Long> cacheModeMemoryVariablePerEx)Reports the cached/cacheable memory requirements.LayerMemoryReport.BuilderLayerMemoryReport.Builder. workingMemory(long fixedInference, long variableInferencePerEx, Map<CacheMode,Long> fixedTrain, Map<CacheMode,Long> variableTrainPerEx)Report the working memory requirements, for both inference and training. -
Uses of CacheMode in org.deeplearning4j.nn.graph
Methods in org.deeplearning4j.nn.graph with parameters of type CacheMode Modifier and Type Method Description voidComputationGraph. setCacheMode(CacheMode mode)This method sets specified CacheMode for all layers within network -
Uses of CacheMode in org.deeplearning4j.nn.layers
Fields in org.deeplearning4j.nn.layers declared as CacheMode Modifier and Type Field Description protected CacheModeAbstractLayer. cacheModeMethods in org.deeplearning4j.nn.layers with parameters of type CacheMode Modifier and Type Method Description voidAbstractLayer. setCacheMode(CacheMode mode)voidFrozenLayer. setCacheMode(CacheMode mode) -
Uses of CacheMode in org.deeplearning4j.nn.layers.recurrent
Methods in org.deeplearning4j.nn.layers.recurrent with parameters of type CacheMode Modifier and Type Method Description static FwdPassReturnLSTMHelpers. activateHelper(BaseRecurrentLayer layer, NeuralNetConfiguration conf, IActivation gateActivationFn, INDArray input, INDArray recurrentWeights, INDArray originalInputWeights, INDArray biases, boolean training, INDArray originalPrevOutputActivations, INDArray originalPrevMemCellState, boolean forBackprop, boolean forwards, String inputWeightKey, INDArray maskArray, boolean hasPeepholeConnections, LSTMHelper helper, CacheMode cacheMode, LayerWorkspaceMgr workspaceMgr, boolean isHelperAllowFallback)Returns FwdPassReturn object with activations/INDArrays.voidBidirectionalLayer. setCacheMode(CacheMode mode) -
Uses of CacheMode in org.deeplearning4j.nn.layers.variational
Fields in org.deeplearning4j.nn.layers.variational declared as CacheMode Modifier and Type Field Description protected CacheModeVariationalAutoencoder. cacheModeMethods in org.deeplearning4j.nn.layers.variational with parameters of type CacheMode Modifier and Type Method Description voidVariationalAutoencoder. setCacheMode(CacheMode mode) -
Uses of CacheMode in org.deeplearning4j.nn.layers.wrapper
Methods in org.deeplearning4j.nn.layers.wrapper with parameters of type CacheMode Modifier and Type Method Description voidBaseWrapperLayer. setCacheMode(CacheMode mode) -
Uses of CacheMode in org.deeplearning4j.nn.multilayer
Methods in org.deeplearning4j.nn.multilayer with parameters of type CacheMode Modifier and Type Method Description voidMultiLayerNetwork. setCacheMode(CacheMode mode)This method sets specified CacheMode for all layers within network
-