NeuralNetworkExecutionContext
Context for the DSL to define the data type and operations.
This class holds the information about the data type and operations that should be used in the DSL. It's used to make the DSL generic and to avoid hardcoding the data type.
Inheritors
Functions
Link copied to clipboard
inline fun <T : DType, V> NeuralNetworkExecutionContext.network(content: NeuralNetworkDsl<T, V>.() -> Unit): Module<T, V>
inline fun <T : DType, V> NeuralNetworkExecutionContext.network(executionContext: ExecutionContext, content: NeuralNetworkDsl<T, V>.() -> Unit): Module<T, V>
Extension function to create a network within a NetworkContext. This bridges the context wrapper with the network DSL using the context's tensor factory.