Versions in this module Expand all Collapse all v0 v0.3.0 Jul 21, 2025 v0.2.0 Jun 26, 2025 Changes in this version + type CoreBuilder interface + Argument func(name string, shape *shape.Shape, index int) (Node, error) + Binary func(op *ast.BinaryExpr, x, y Node) (Node, error) + BroadcastInDim func(x Node, shape *shape.Shape, broadcastAxes []int) (Node, error) + Call func(sg Subgraph, args ...Node) (Node, error) + Cast func(x Node, target dtype.DataType) (Node, error) + Concat func(axis int, nodes []Node) (Node, error) + Constant func(value platform.HostBuffer) (Node, error) + DotGeneral func(x, y Node, batchAxes, reduceAxes [2][]int) (Node, error) + Graph func() Graph + Reshape func(x Node, axisLengths []int) (Node, error) + Set func(x, updates, index Node) (Node, error) + Slice func(x Node, index int) (Node, error) + Subgraph func(name string) (Graph, error) + Tuple func(nodes []Node) (Tuple, error) + Unary func(op *ast.UnaryExpr, x Node) (Node, error) + While func(cond, body Subgraph, state Node) (Node, error) + type Graph interface + Compile func(dev platform.Device, output, traced []*OutputNode, params []*shape.Shape) (Runner, error) + Core func() CoreBuilder + Math func() MathBuilder + Num func() NumBuilder + Platform func() platform.Platform + type MathBuilder interface + Cos func(x Node) (Node, error) + Sin func(x Node) (Node, error) + Tanh func(x Node) (Node, error) + type Node interface + Graph func() Graph + type NumBuilder interface + Iota func(sh *shape.Shape, iotaAxis int) (Node, error) + type OutputNode struct + Node Node + Shape *shape.Shape + func (out *OutputNode) String() string + type Runner interface + Run func([]platform.Handle) (out, traces []platform.DeviceHandle, err error) + type Subgraph struct + Graph Graph + Result OutputNode + type Tuple interface + Element func(i int) (Node, error) + Size func() int + Unpack func() ([]Node, error)
Go to main page