skainet-compile-core/sk.ainet.tape/ExecutionTape

ExecutionTape

Interface for recording operations during execution for deferred/graph-based execution. The tape captures operations as they are performed, allowing for later replay or optimization.

Inheritors

GradientTape

SimpleExecutionTape

Properties

isRecording

abstract val isRecording: Boolean

Whether this tape is currently recording operations

operations

abstract val operations: List<RecordedOperation>

All recorded operations on this tape

Functions

clear

abstract fun clear()

Clear all recorded operations

copy

abstract fun copy(): ExecutionTape

Create a copy of this tape

optimize

abstract fun optimize(): ExecutionTape

Optimize the recorded operations (e.g., operation fusion, dead code elimination)

prune

abstract fun prune(keepOutputs: Set<String> = emptySet()): ExecutionTape

Prune unnecessary operations from the tape

recordOperation

abstract fun <T : DType, V> recordOperation(operation: Operation, inputs: List<Tensor<T, V>>, outputs: List<Tensor<T, V>>)

Record an operation on this tape

replay

abstract fun <T : DType, V> replay(): List<Tensor<T, V>>

Replay all recorded operations

startRecording

abstract fun startRecording()

Start recording operations on this tape

stopRecording

abstract fun stopRecording()

Stop recording operations on this tape