Uses of Interface
io.github.kirstenali.deepj.tensor.TensorBackend
Packages that use TensorBackend
Package
Description
-
Uses of TensorBackend in io.github.kirstenali.deepj.tensor
Subinterfaces of TensorBackend in io.github.kirstenali.deepj.tensorModifier and TypeInterfaceDescriptioninterfaceOptional backend capability for memory-efficient, fused causal attention.Methods in io.github.kirstenali.deepj.tensor that return TensorBackendMethods in io.github.kirstenali.deepj.tensor with parameters of type TensorBackendModifier and TypeMethodDescriptionstatic CrossEntropyResultCrossEntropyDefaults.apply(TensorBackend backend, Tensor logits, int[] targets, boolean[] mask) static TensorPackedLayouts.matmul(TensorBackend backend, Tensor left, Tensor right, int batches, boolean transposeLeft, boolean transposeRight) static TensorPackedLayouts.merge(TensorBackend backend, Tensor input, int heads) static TensorHeadCacheOps.prefix(TensorBackend backend, Tensor cache, int heads, int length) static voidTensor.setBackend(TensorBackend backend) static TensorPackedLayouts.split(TensorBackend backend, Tensor input, int heads) static voidHeadCacheOps.writeLayouts(TensorBackend backend, Tensor cache, Tensor update, int heads, int offset) -
Uses of TensorBackend in io.github.kirstenali.deepj.tensor.cpu
Classes in io.github.kirstenali.deepj.tensor.cpu that implement TensorBackend -
Uses of TensorBackend in io.github.kirstenali.deepj.tensor.metal
Classes in io.github.kirstenali.deepj.tensor.metal that implement TensorBackend