package caisar

  1. Overview
  2. Docs
package caisar
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

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
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct