Class NativeTensorOperations
java.lang.Object
com.github.tjake.jlama.tensor.operations.NativeTensorOperations
- All Implemented Interfaces:
TensorOperations
-
Field Summary
FieldsFields inherited from interface com.github.tjake.jlama.tensor.operations.TensorOperations
scratch -
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionvoidaccumulate(AbstractTensor a, AbstractTensor b, int offset, int length) voidbatchDotProduct(AbstractTensor result, AbstractTensor at, AbstractTensor bt, int aColumnOffset, int bColumnOffset, int columnLength, int rRowOffset, int bRowOffset, int rowChunkSize) voiddotProductBatchChunk(AbstractTensor[] r, AbstractTensor a, AbstractTensor[] b, int columnOffset, int columnLength, int bRowOffset, int rowChunkSize) voidmaccumulate(AbstractTensor a, AbstractTensor b, int offset, int length) name()intquantize(AbstractTensor t, DType qtype, int offset, int length) voidsaxpy(float alpha, AbstractTensor x, AbstractTensor y, int xoffset, int yoffset, int limit) voidsaxpy(AbstractTensor alpha, AbstractTensor x, AbstractTensor y, int xoffset, int yoffset, int limit, int rOffset, int xOffset, int batchSize) voidscale(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, waitMethods 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
- Specified by:
namein interfaceTensorOperations
-
parallelSplitSize
public int parallelSplitSize()- Specified by:
parallelSplitSizein interfaceTensorOperations
-
batchDotProduct
public void batchDotProduct(AbstractTensor result, AbstractTensor at, AbstractTensor bt, int aColumnOffset, int bColumnOffset, int columnLength, int rRowOffset, int bRowOffset, int rowChunkSize) - Specified by:
batchDotProductin interfaceTensorOperations
-
dotProductBatchChunk
public void dotProductBatchChunk(AbstractTensor[] r, AbstractTensor a, AbstractTensor[] b, int columnOffset, int columnLength, int bRowOffset, int rowChunkSize) - Specified by:
dotProductBatchChunkin interfaceTensorOperations
-
accumulate
- Specified by:
accumulatein interfaceTensorOperations
-
maccumulate
- Specified by:
maccumulatein interfaceTensorOperations
-
saxpy
public void saxpy(float alpha, AbstractTensor x, AbstractTensor y, int xoffset, int yoffset, int limit) - Specified by:
saxpyin interfaceTensorOperations
-
saxpy
public void saxpy(AbstractTensor alpha, AbstractTensor x, AbstractTensor y, int xoffset, int yoffset, int limit, int rOffset, int xOffset, int batchSize) - Specified by:
saxpyin interfaceTensorOperations
-
scale
- Specified by:
scalein interfaceTensorOperations
-
quantize
- Specified by:
quantizein interfaceTensorOperations
-