class

ConstructionKit::ExecutableGraph

Inherits Reference < Object

Constructors

Instance methods

add_edge(from_node : String, from_port : String, to_node : String, to_port : String)
Source
add_node(node : ExecutableNode)
Source
all_adam_mats
Source
all_weight_mats
Source
backward(lr : Float64)
Source
compute_topo_order!

Compute topological order using Kahn's algorithm

Source
edges
Source
forward(boundary_inputs : Hash(String, Tensor)) : Float64
Source
generate(seed_ids : Array(Int32), max_tokens : Int32, temperature : Float64, seq_len : Int32) : Array(Int32)
Source
loss_node_id
Source
nodes
Source
topo_order
Source
total_params
Source
train_step(input_ids : Array(Int32), target_ids : Array(Int32), lr : Float64) : Float64
Source
warnings
Source