Index

A B C D E F G H I L M N O P R S T U V W X Y Z
All Classes and Interfaces|All Packages|Constant Field Values

R

random(int, int, Random) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

random(int, int, Random) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

random(int, int, Random) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

random(int, int, Random) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

random(Random) - Method in class io.github.kirstenali.deepj.transformer.TransformerBuilder

recordAdamWUpdate(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, float, float, float, float, float, float, float, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record in-place AdamW update: [OP_ADAMW_UPDATE, wId, gId, mtId, vtId, lrBits, beta1Bits, beta2Bits, epsBits, weightDecayBits, bc1Bits, bc2Bits, n]

recordBinary(int, GpuBuffer, GpuBuffer, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record a binary element-wise op: [opCode, aId, bId, outId, n]

recordLayerNormBackward(GpuBuffer, GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record layer norm backward: [OP_LAYERNORM_BACKWARD, dXHatId, xHatId, stdId, outId, rows, cols]

recordMatmul(GpuBuffer, GpuBuffer, GpuBuffer, int, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record matmul: [OP_MATMUL, aId, bId, outId, m, n, k]

recordMultiplyScalar(GpuBuffer, GpuBuffer, float) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record scalar multiply: [OP_MULTIPLY_SCALAR, inId, outId, scalarBits, n]

recordSoftmaxBackward(GpuBuffer, GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record softmax backward: [OP_SOFTMAX_BACKWARD, gradId, softmaxId, outId, rows, cols]

recordSoftmaxRows(GpuBuffer, GpuBuffer, int, int) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record softmax rows: [OP_SOFTMAX_ROWS, inId, outId, rows, cols]

recordUnary(int, GpuBuffer, GpuBuffer) - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Record a unary op: [opCode, inId, outId, n]

releaseAll() - Method in class io.github.kirstenali.deepj.tensor.ComputeGraph

Release all GPU buffers and reset the graph completely.

releaseBuffers(int[], int) - Method in interface io.github.kirstenali.deepj.tensor.GpuRuntime

Release multiple GPU buffers by id.

releaseResources() - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

releaseResources() - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

Release backend-owned resources (GPU buffers, native handles, etc.).

relu(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

relu(Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

relu(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

ReLU - Class in io.github.kirstenali.deepj.activations

ReLU() - Constructor for class io.github.kirstenali.deepj.activations.ReLU

reluActivation() - Method in class io.github.kirstenali.deepj.tensor.Tensor

reluBackward(Tensor) - Method in class io.github.kirstenali.deepj.tensor.Tensor

reluBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

reluBackward(Tensor, Tensor) - Method in class io.github.kirstenali.deepj.tensor.metal.MetalBackend

reluBackward(Tensor, Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

reluInPlace() - Method in class io.github.kirstenali.deepj.tensor.Tensor

reluInPlace(Tensor) - Method in class io.github.kirstenali.deepj.tensor.cpu.CpuBackend

reluInPlace(Tensor) - Method in interface io.github.kirstenali.deepj.tensor.TensorBackend

requireSameShape(Tensor, Tensor, String) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

requireTargetsMatchRows(Tensor, int[]) - Static method in class io.github.kirstenali.deepj.tensor.Tensor

right() - Method in record class io.github.kirstenali.deepj.tokenizers.bpe.TokenPair

Returns the value of the right record component.

rows - Variable in class io.github.kirstenali.deepj.tensor.GpuBuffer

rows - Variable in class io.github.kirstenali.deepj.tensor.Tensor

A B C D E F G H I L M N O P R S T U V W X Y Z
All Classes and Interfaces|All Packages|Constant Field Values