FusedOpHandler

fun interface FusedOpHandler<T : DType, V>(source)

Handler for executing a fused operation.

Backends implement this to provide platform-specific kernels for fused ops. For example, a Metal backend would register handlers for fused_rms_norm and fused_swiglu_ffn that dispatch to Metal Performance Shaders.

Functions

Link copied to clipboard
abstract fun execute(ops: TensorOps, inputs: List<Tensor<T, V>>, params: Map<String, Any>): List<Tensor<T, V>>

Execute the fused operation.