package caisar

  1. Overview
  2. Docs

Nodes

Computation graphs are made up of a DAG of nodes, which represent what is commonly called a "layer" or "pipeline stage" in machine learning frameworks.

For example, it can be a node of type "Conv" that takes in an image, a filter tensor and a bias tensor, and produces the convolved output.

type t = {
  1. input : string list;
    (*

    namespace Value

    *)
  2. output : string list;
    (*

    namespace Value

    *)
  3. name : string option;
    (*

    An optional identifier for this node in a graph. This field MAY be absent in ths version of the IR.

    namespace Node

    *)
  4. op_type : string option;
    (*

    The symbolic identifier of the Operator to execute.

    namespace Operator

    *)
  5. attribute : AttributeProto.t list;
    (*

    Additional named attributes.

    *)
  6. doc_string : string option;
    (*

    A human-readable documentation for this node. Markdown is allowed.

    *)
  7. domain : string option;
    (*

    The domain of the OperatorSet that specifies the operator named by op_type.

    namespace Domain

    *)
}
val make : ?input:string list -> ?output:string list -> ?name:string -> ?op_type:string -> ?attribute:AttributeProto.t list -> ?doc_string:string -> ?domain:string -> unit -> t

Helper function to generate a message using default values

Serialize the message to binary format

Deserialize from binary format

Serialize to Json (compatible with Yojson.Basic.t)

Deserialize from Json (compatible with Yojson.Basic.t)

val name : unit -> string

Fully qualified protobuf name of this message

OCaml

Innovation. Community. Security.