package caisar

You can search for identifiers within the package.

in-package search v0.2.0

On This Page

Legend:
Library
Module
Module type
Parameter
Class
Class type

Neural Intermediate Representation (NIR)

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

A graph is described starting from its output node.

type t

val pp : t Fmt.t

val pp_debug : t Fmt.t

val create : Node.t -> t

Create a network from its output node.t

val output : t -> Node.t

Output node.t of the network

val nodes : t -> Node.t list

Output nodes of the network

val input_shape : t -> Shape.t

Input shape of the network

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

successors of a node.t

val iter_vertex : (Node.t -> unit) -> t -> unit

val iter_succ : (Node.t -> unit) -> t -> Node.t -> unit

val grapheasy : t -> string

module GFloat : sig ... end

Respect some OcamlGraph signature

module Dot : sig ... end