caisar 2.0 · OCaml Package

Graphs

A graph defines the computational logic of a model and is comprised of a parameterized list of nodes that form a directed acyclic graph based on their inputs and outputs. This is the equivalent of the "network" or "graph" in many deep learning frameworks.

type t = {

node : NodeProto.t list;
(*
The nodes in the graph, sorted topologically.
*)
name : string option;
(*
The name of the graph.
namespace Graph
*)
initializer' : TensorProto.t list;
(*
A list of named tensor values, used to specify constant inputs of the graph. Each initializer (both TensorProto as well SparseTensorProto) MUST have a name. The name MUST be unique across both initializer and sparse_initializer, but the name MAY also appear in the input list.
*)
doc_string : string option;
(*
A human-readable documentation for this graph. Markdown is allowed.
*)
input : ValueInfoProto.t list;
(*
The inputs and outputs of the graph.
*)
output : ValueInfoProto.t list;
value_info : ValueInfoProto.t list;
(*
Information for the values in the graph. The ValueInfoProto.name's must be distinct. It is optional for a value to appear in value_info list.
*)
quantization_annotation : TensorAnnotation.t list;
(*
This field carries information to indicate the mapping among a tensor and its quantization parameter tensors. For example: For tensor 'a', it may have {'SCALE_TENSOR', 'a_scale'} and {'ZERO_POINT_TENSOR', 'a_zero_point'} annotated, which means, tensor 'a_scale' and tensor 'a_zero_point' are scale and zero point of tensor 'a' in the model.
*)
sparse_initializer : SparseTensorProto.t list;
(*
Initializers (see above) stored in sparse format.
*)

}

val make : 
  ?node:NodeProto.t list ->
  ?name:string ->
  ?initializer':TensorProto.t list ->
  ?doc_string:string ->
  ?input:ValueInfoProto.t list ->
  ?output:ValueInfoProto.t list ->
  ?value_info:ValueInfoProto.t list ->
  ?quantization_annotation:TensorAnnotation.t list ->
  ?sparse_initializer:SparseTensorProto.t list ->
  unit ->
  t

Helper function to generate a message using default values

val to_proto : t -> Ocaml_protoc_plugin.Writer.t

Serialize the message to binary format

val from_proto : 
  Ocaml_protoc_plugin.Reader.t ->
  (t, [> Ocaml_protoc_plugin.Result.error ]) result

Deserialize from binary format

val to_json : 
  Ocaml_protoc_plugin.Json_options.t ->
  t ->
  Ocaml_protoc_plugin.Json.t

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

val from_json : 
  Ocaml_protoc_plugin.Json.t ->
  (t, [> Ocaml_protoc_plugin.Result.error ]) result

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

val name : unit -> string

Fully qualified protobuf name of this message

package caisar