Package | Description |
---|---|
org.nd4j.jita.allocator.utils | |
org.nd4j.linalg.jcublas | |
org.nd4j.linalg.jcublas.buffer |
Modifier and Type | Method and Description |
---|---|
static AllocationShape |
AllocationUtils.buildAllocationShape(JCudaBuffer buffer)
This method returns AllocationShape for specific buffer, that takes in account its real shape: offset, length, etc
|
Modifier and Type | Method and Description |
---|---|
JCudaBuffer |
CublasPointer.getBuffer() |
Constructor and Description |
---|
CublasPointer(JCudaBuffer buffer,
CudaContext context)
Creates a CublasPointer
for a given JCudaBuffer
|
Modifier and Type | Class and Description |
---|---|
class |
BaseCudaDataBuffer
Base class for a data buffer
CUDA implementation for DataBuffer always uses JavaCPP as allocationMode, and device access is masked by appropriate allocator mover implementation.
|
class |
CudaDoubleDataBuffer
Cuda double buffer
|
class |
CudaFloatDataBuffer
Cuda float buffer
|
class |
CudaIntDataBuffer
Cuda int buffer
|
Copyright © 2016. All Rights Reserved.