Module Nir.NgraphSource

Neural Intermediate Representation (NIR)

NIR is a graph describing a machine learning model control flow.

A graph is described starting from its output node.

Sourcetype t
Sourceval pp : t Fmt.t
Sourceval pp_debug : t Fmt.t
Sourceval create : Node.t -> t

Create a network from its output node.t

Sourceval output : t -> Node.t

Output node.t of the network

Sourceval nodes : t -> Node.t list

Output nodes of the network

Sourceval input_shape : t -> Shape.t

Input shape of the network

Sourceval succs : t -> Node.t -> Node.t list

successors of a node.t

Sourceval iter_vertex : (Node.t -> unit) -> t -> unit
Sourceval iter_succ : (Node.t -> unit) -> t -> Node.t -> unit
Sourceval grapheasy : t -> string
Sourcemodule GFloat : sig ... end

Respect some OcamlGraph signature

Sourcemodule Dot : sig ... end