skainet-compile-dag/sk.ainet.lang.graph/DefaultExecutionTape

DefaultExecutionTape

open class DefaultExecutionTape(var session: TraceSession = sk.ainet.lang.trace.TraceSession()) : ExecutionTape(source)

Default implementation of ExecutionTape

Inheritors

DefaultGradientTape

Constructors

DefaultExecutionTape

constructor(session: TraceSession = sk.ainet.lang.trace.TraceSession())

Properties

open override val isRecording: Boolean

open override val operations: List<RecordedOperation>

recordingStrategy

var recordingStrategy: TapeRecordingStrategy

var session: TraceSession

val sessionRef: TraceSession

val traces: List<OpTrace>

Functions

open override fun clear()

open override fun copy(): ExecutionTape

open override fun optimize(): ExecutionTape

open override fun prune(keepOutputs: Set<String>): ExecutionTape

recordOperation

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

open fun recordTrace(trace: OpTrace)

Record a high-level OpTrace into this tape (used by TapeSink).

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

open override fun startRecording()

open override fun stopRecording()

fun toComputeGraph(synthesizeExternalInputs: Boolean = false, inputTensorIds: Set<String> = emptySet()): ComputeGraph