Package-level declarations

Types

Link copied to clipboard
interface GraphDslItem
Link copied to clipboard
Link copied to clipboard
Link copied to clipboard
data class GraphExecutionResult<V>(val result: V, val graph: ComputeGraph, val tape: ExecutionTape)

Result of graph execution containing the computed result and execution metadata

Link copied to clipboard

DSL Scope for Skainet that wraps a GraphExecutionContext and provides a more idiomatic way to interact with tensors and training loops.

Functions

Link copied to clipboard
fun <V, R> compileGraphExec(graphExecutionContext: GraphExecutionContext, block: GraphExecContextDsl.(executionContext: GraphExecutionContext) -> R): GraphExecutionResult<R>

Creates a graph execution context and executes the given block within it. Similar to the network context pattern but for graph execution.

Link copied to clipboard

Entry point for the Skainet DSL.

Link copied to clipboard
fun GraphProgram.toComputeGraph(into: ComputeGraph = DefaultComputeGraph()): ComputeGraph

Lower a symbolic GraphProgram (produced by the DAG DSL) into a concrete ComputeGraph.