skainet-lang-ksp-annotations/sk.ainet.lang.ops/TensorOp

TensorOp

@Target(allowedTargets = [AnnotationTarget.FUNCTION])

annotation class TensorOp(val mode: ComputationMode = ComputationMode.INFERENCE)(source)

Annotation for functions that should be processed by the Mikrograd KSP processor. The processor will generate optimized code for the function based on the computation mode.

Parameters

mode

The computation mode to use (INFERENCE or TRAINING)

Properties

mode

val mode: ComputationMode