Package | Description |
---|---|
org.nd4j.jita.allocator.impl | |
org.nd4j.jita.handler | |
org.nd4j.jita.handler.impl | |
org.nd4j.jita.memory | |
org.nd4j.jita.memory.impl |
Modifier and Type | Method and Description |
---|---|
PointersPair |
AllocationPoint.getPointers() |
Modifier and Type | Method and Description |
---|---|
void |
AllocationPoint.setPointers(PointersPair pointerInfo)
This method sets CUDA pointer for this allocation.
|
Modifier and Type | Method and Description |
---|---|
PointersPair |
MemoryHandler.alloc(AllocationStatus targetMode,
AllocationPoint point,
AllocationShape shape,
boolean initialize)
Allocate specified memory chunk on specified device/host
|
Modifier and Type | Method and Description |
---|---|
PointersPair |
CudaZeroHandler.alloc(AllocationStatus targetMode,
AllocationPoint point,
AllocationShape shape,
boolean initialize)
Allocate specified memory chunk on specified device/host
|
Modifier and Type | Method and Description |
---|---|
PointersPair |
MemoryProvider.malloc(AllocationShape shape,
AllocationPoint point,
AllocationStatus location)
This method provides PointersPair to memory chunk specified by AllocationShape
|
Modifier and Type | Method and Description |
---|---|
PointersPair |
CudaFullCachingProvider.malloc(AllocationShape shape,
AllocationPoint point,
AllocationStatus location) |
PointersPair |
CudaDirectProvider.malloc(AllocationShape shape,
AllocationPoint point,
AllocationStatus location)
This method provides PointersPair to memory chunk specified by AllocationShape
|
PointersPair |
CudaCachingZeroProvider.malloc(AllocationShape shape,
AllocationPoint point,
AllocationStatus location)
This method provides PointersPair to memory chunk specified by AllocationShape
PLEASE NOTE: This method can actually ignore malloc request, and give out previously cached free memory chunk with equal shape.
|
Copyright © 2016. All Rights Reserved.