skainet-lang-dag/sk.ainet.lang.dag

Package-level declarations

Types

DagBuilder

class DagBuilder

Builder that records symbolic nodes/values. It is definition-only: no tensors are allocated.

DagDsl

@DslMarker

annotation class DagDsl

DagModule

abstract class DagModule

A reusable sub-graph module that can be instantiated within a DagBuilder.

GraphDslOps

interface GraphDslOps : TensorOps

GraphNodeDefinition

data class GraphNodeDefinition(val id: String, val operation: Operation, val inputs: List<GraphValue<*>>, val outputs: List<GraphValue<*>>, val attributes: Map<String, Any?> = emptyMap())

Logical node definition captured by the DSL before lowering to sk.ainet.lang.graph.ComputeGraph.

GraphProgram

data class GraphProgram(val nodes: List<GraphNodeDefinition>, val outputs: List<GraphValue<*>>)

Immutable program emitted by the DSL.

GraphValue

data class GraphValue<out T : DType>(val nodeId: String, val outputIndex: Int, val spec: TensorSpec)

Symbolic value flowing through the DAG DSL. Every value is produced by a node output.

SymbolicInit

class SymbolicInit

Records a symbolic initializer hint (used only as metadata on TensorSpec).

SymbolicTensorBuilder

class SymbolicTensorBuilder<T : DType>(dtype: KClass<T>, defaultName: String)

Lightweight, allocation-free builder that mimics the shape/initializer style of the data DSL but produces only TensorSpec metadata for the DAG DSL.

Functions

abs

fun DagBuilder.abs(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for abs.

add

fun DagBuilder.add(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for add.

addScalar

fun DagBuilder.addScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for addScalar.

avgPool2d

fun DagBuilder.avgPool2d(input: GraphValue<*>, kernelSize: Pair<Int, Int>, stride: Pair<Int, Int> = 1 to 1, padding: Pair<Int, Int> = 0 to 0, countIncludePad: Boolean, id: String = ""): GraphValue<*>

DSL extension for avgPool2d.

clamp

fun DagBuilder.clamp(tensor: GraphValue<*>, minVal: Float, maxVal: Float, id: String = ""): GraphValue<*>

DSL extension for clamp.

concat

fun DagBuilder.concat(tensors: List<GraphValue<*>>, dim: Int, id: String = ""): GraphValue<*>

DSL extension for concat.

conv1d

fun DagBuilder.conv1d(input: GraphValue<*>, weight: GraphValue<*>, bias: GraphValue<*>, stride: Int, padding: Int, dilation: Int, groups: Int = 1, id: String = ""): GraphValue<*>

DSL extension for conv1d.

conv2d

fun DagBuilder.conv2d(input: GraphValue<*>, weight: GraphValue<*>, bias: GraphValue<*>, stride: Pair<Int, Int> = 1 to 1, padding: Pair<Int, Int> = 0 to 0, dilation: Pair<Int, Int> = 1 to 1, groups: Int = 1, id: String = ""): GraphValue<*>

DSL extension for conv2d.

conv3d

fun DagBuilder.conv3d(input: GraphValue<*>, weight: GraphValue<*>, bias: GraphValue<*>, stride: Triple<Int, Int, Int>, padding: Triple<Int, Int, Int>, dilation: Triple<Int, Int, Int>, groups: Int = 1, id: String = ""): GraphValue<*>

DSL extension for conv3d.

dag

fun dag(block: DagBuilder.() -> Unit): GraphProgram

Entry point for the DAG DSL.

dagModule

fun dagModule(block: DagBuilder.(List<GraphValue<*>>) -> List<GraphValue<*>>): DagModule

Defines a reusable graph component (module).

divide

fun DagBuilder.divide(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for divide.

divScalar

fun DagBuilder.divScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for divScalar.

elu

fun DagBuilder.elu(tensor: GraphValue<*>, alpha: Float, id: String = ""): GraphValue<*>

DSL extension for elu.

flatten

fun DagBuilder.flatten(tensor: GraphValue<*>, startDim: Int = 0, endDim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for flatten.

fun DagBuilder.ge(tensor: GraphValue<*>, value: Float, id: String = ""): GraphValue<*>

DSL extension for ge.

gelu

fun DagBuilder.gelu(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for gelu.

leakyRelu

fun DagBuilder.leakyRelu(tensor: GraphValue<*>, negativeSlope: Float, id: String = ""): GraphValue<*>

DSL extension for leakyRelu.

logSoftmax

fun DagBuilder.logSoftmax(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for logSoftmax.

fun DagBuilder.lt(tensor: GraphValue<*>, value: Float, id: String = ""): GraphValue<*>

DSL extension for lt.

matmul

fun DagBuilder.matmul(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for matmul.

maxPool2d

fun DagBuilder.maxPool2d(input: GraphValue<*>, kernelSize: Pair<Int, Int>, stride: Pair<Int, Int> = 1 to 1, padding: Pair<Int, Int> = 0 to 0, id: String = ""): GraphValue<*>

DSL extension for maxPool2d.

mean

fun DagBuilder.mean(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for mean.

module

fun DagBuilder.module(module: DagModule, inputs: List<GraphValue<*>>): List<GraphValue<*>>

DSL helper for using modules.

mulScalar

fun DagBuilder.mulScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for mulScalar.

multiply

fun DagBuilder.multiply(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for multiply.

narrow

fun DagBuilder.narrow(tensor: GraphValue<*>, dim: Int, start: Int, length: Int, id: String = ""): GraphValue<*>

DSL extension for narrow.

pad2d

fun DagBuilder.pad2d(tensor: GraphValue<*>, padLeft: Int, padRight: Int, padTop: Int, padBottom: Int, id: String = ""): GraphValue<*>

DSL extension for pad2d.

rdivScalar

fun DagBuilder.rdivScalar(a: Number, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for rdivScalar.

relu

fun DagBuilder.relu(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for relu.

reshape

fun DagBuilder.reshape(tensor: GraphValue<*>, newShape: Shape, id: String = ""): GraphValue<*>

DSL extension for reshape.

rsubScalar

fun DagBuilder.rsubScalar(a: Number, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for rsubScalar.

sigmoid

fun DagBuilder.sigmoid(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for sigmoid.

sign

fun DagBuilder.sign(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for sign.

silu

fun DagBuilder.silu(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for silu.

softmax

fun DagBuilder.softmax(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for softmax.

sqrt

fun DagBuilder.sqrt(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for sqrt.

squeeze

fun DagBuilder.squeeze(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for squeeze.

subScalar

fun DagBuilder.subScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for subScalar.

subtract

fun DagBuilder.subtract(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for subtract.

sum

fun DagBuilder.sum(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for sum.

transpose

fun DagBuilder.transpose(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for transpose.

tril

fun DagBuilder.tril(tensor: GraphValue<*>, k: Int, id: String = ""): GraphValue<*>

DSL extension for tril.

unfold

fun DagBuilder.unfold(tensor: GraphValue<*>, dim: Int, size: Int, step: Int, id: String = ""): GraphValue<*>

DSL extension for unfold.

unsqueeze

fun DagBuilder.unsqueeze(tensor: GraphValue<*>, dim: Int, id: String = ""): GraphValue<*>

DSL extension for unsqueeze.

upsample2d

fun DagBuilder.upsample2d(input: GraphValue<*>, scale: Pair<Int, Int>, mode: UpsampleMode = UpsampleMode.Nearest, alignCorners: Boolean = false, id: String = ""): GraphValue<*>

DSL extension for upsample2d.

variance

fun DagBuilder.variance(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for variance.