skainet-lang-core/sk.ainet.context.observers/MemorySnapshotObserver

MemorySnapshotObserver

class MemorySnapshotObserver : ExecutionObserver, ResettableExecutionObserver(source)

Captures the context's reported memory usage after each operation.

Constructors

MemorySnapshotObserver

constructor()

Functions

open override fun onOpEnd(context: ExecutionContext, opName: String, result: Any?)

Called after a successful operation. result can be a single tensor, a collection of tensors, or any backend specific payload.

open override fun reset()

fun results(): List<MemorySample>