Package io.github.kirstenali.deepj.tensor

Class GpuBuffer

java.lang.Object

io.github.kirstenali.deepj.tensor.GpuBuffer

public final class GpuBuffer
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
boolean	allocatedOnGpu
final int	cols
boolean	cpuStale
final DType	dtype
final int	id
boolean	needsUpload
final int	rows

Constructor Summary

Constructors

Constructor	Description
GpuBuffer(int id, int rows, int cols, boolean needsUpload)
GpuBuffer(int id, int rows, int cols, DType dtype, boolean needsUpload)

Method Summary

Modifier and Type	Method	Description
long	byteLength()
int	elementCount()
int	floatCount()

Methods inherited from class java.lang.Object

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

Field Details

id

public final int id

rows

public final int rows

cols

public final int cols

dtype

public final DType dtype

needsUpload

public boolean needsUpload

cpuStale

public boolean cpuStale

allocatedOnGpu

public boolean allocatedOnGpu

Constructor Details

GpuBuffer

public GpuBuffer(int id,
 int rows,
 int cols,
 boolean needsUpload)

GpuBuffer

public GpuBuffer(int id,
 int rows,
 int cols,
 DType dtype,
 boolean needsUpload)

Method Details

elementCount

public int elementCount()

byteLength

public long byteLength()

floatCount

public int floatCount()