DagBuilder

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

Constructors

constructor()

Constant helper that reuses a symbolic, allocation-free data DSL to declare shape/dtype.

Declare a constant placeholder (treated like an input node).

fun <T : DType> input(name: String, spec: TensorSpec = TensorSpec(name = name, shape = null, dtype = "unknown")): GraphValue<T>

Declare a graph input placeholder.

DSL helper for using modules.

fun op(operation: Operation, inputs: List<GraphValue<*>>, id: String = "", attributes: Map<String, Any?> = emptyMap()): List<GraphValue<*>>

Generic operation hook that lets callers wire custom Operation instances.

fun output(vararg values: GraphValue<*>)

Mark a value as a program output. If none are marked, the last node's outputs are used.

Parameter helper that reuses a symbolic, allocation-free data DSL to declare shape/dtype.

Declare a parameter/weight placeholder.