Package com.github.tjake.jlama.tensor.operations

Class NativeTensorOperations

java.lang.Object

com.github.tjake.jlama.tensor.operations.NativeTensorOperations

All Implemented Interfaces:

TensorOperations

public class NativeTensorOperations
extends Object
implements TensorOperations

Field Summary

Fields

Modifier and Type	Field	Description
static final int	HAS_AVX2
static final int	HAS_F16C

Fields inherited from interface com.github.tjake.jlama.tensor.operations.TensorOperations

scratch

Constructor Summary

Constructors

Constructor	Description
NativeTensorOperations()

Method Summary

Modifier and Type	Method	Description
void	accumulate(AbstractTensor a, AbstractTensor b, int offset, int length)
void	batchDotProduct(AbstractTensor result, AbstractTensor at, AbstractTensor bt, int aColumnOffset, int bColumnOffset, int columnLength, int rRowOffset, int bRowOffset, int rowChunkSize)
void	dotProductBatchChunk(AbstractTensor[] r, AbstractTensor a, AbstractTensor[] b, int columnOffset, int columnLength, int bRowOffset, int rowChunkSize)
void	maccumulate(AbstractTensor a, AbstractTensor b, int offset, int length)
String	name()
int	parallelSplitSize()
AbstractTensor	quantize(AbstractTensor t, DType qtype, int offset, int length)
void	saxpy(float alpha, AbstractTensor x, AbstractTensor y, int xoffset, int yoffset, int limit)
void	saxpy(AbstractTensor alpha, AbstractTensor x, AbstractTensor y, int xoffset, int yoffset, int limit, int rOffset, int xOffset, int batchSize)
void	scale(float factor, AbstractTensor x, int offset, int length)

Methods inherited from class java.lang.Object

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

Methods inherited from interface com.github.tjake.jlama.tensor.operations.TensorOperations

batchDotProduct, dotProduct, dotProduct, dotProductChunk, sum

Field Details

HAS_F16C

public static final int HAS_F16C

HAS_AVX2

public static final int HAS_AVX2

Constructor Details

NativeTensorOperations

public NativeTensorOperations()

Method Details

name

public String name()

Specified by:

name in interface TensorOperations

parallelSplitSize

public int parallelSplitSize()

Specified by:

parallelSplitSize in interface TensorOperations

batchDotProduct

public void batchDotProduct(AbstractTensor result,
 AbstractTensor at,
 AbstractTensor bt,
 int aColumnOffset,
 int bColumnOffset,
 int columnLength,
 int rRowOffset,
 int bRowOffset,
 int rowChunkSize)

Specified by:

batchDotProduct in interface TensorOperations

dotProductBatchChunk

public void dotProductBatchChunk(AbstractTensor[] r,
 AbstractTensor a,
 AbstractTensor[] b,
 int columnOffset,
 int columnLength,
 int bRowOffset,
 int rowChunkSize)

Specified by:

dotProductBatchChunk in interface TensorOperations

accumulate

public void accumulate(AbstractTensor a,
 AbstractTensor b,
 int offset,
 int length)

Specified by:

accumulate in interface TensorOperations

maccumulate

public void maccumulate(AbstractTensor a,
 AbstractTensor b,
 int offset,
 int length)

Specified by:

maccumulate in interface TensorOperations

saxpy

public void saxpy(float alpha,
 AbstractTensor x,
 AbstractTensor y,
 int xoffset,
 int yoffset,
 int limit)

Specified by:

saxpy in interface TensorOperations

saxpy

public void saxpy(AbstractTensor alpha,
 AbstractTensor x,
 AbstractTensor y,
 int xoffset,
 int yoffset,
 int limit,
 int rOffset,
 int xOffset,
 int batchSize)

Specified by:

saxpy in interface TensorOperations

scale

public void scale(float factor,
 AbstractTensor x,
 int offset,
 int length)

Specified by:

scale in interface TensorOperations

quantize

public AbstractTensor quantize(AbstractTensor t,
 DType qtype,
 int offset,
 int length)

Specified by:

quantize in interface TensorOperations