caisar-ir 0.2 (latest) · OCaml Package

type shape = Base.int Base.array

type operator =

| Add
| Sub
| Mul
| Div
| Matmul
| Gemm
| LogSoftmax
| ReLu
| Transpose
| Squeeze
| MaxPool
| Conv
| Reshape
| Flatten
| Identity
| Constant
| NO_OP
| RW_Linearized_ReLu

Type describing the different operations handled. Those operations are inspired by those defined in the ONNX documentation.

see https://github.com/onnx/onnx/blob/master/docs/Operators.md
for more informations. They are to be coupled with the relevant operators parameters.

val str_op : operator -> Base.string

val show_shape : shape -> Base.string

type ksize =

| Ksize of shape

type stride =

| Stride of shape

type pads =

| Pads of shape

type dilations =

| Dilations of shape

type operator_parameters =

| Pool_params of ksize * stride Base.option * pads Base.option * dilations Base.option
| Conv_params of ksize * stride Base.option * pads Base.option * dilations Base.option
| Transpose_params of shape
| RW_Linearized_ReLu_params of Base.bool Base.list Base.list * ((Base.string, Base.float) Base.Hashtbl.t Base.list * Base.int)

val str_op_params : operator_parameters -> Base.string

type ('a, 'b) t = {

id : Base.int;
name : Base.string Base.option;
shape : shape;
operator : operator;
operator_parameters : operator_parameters Base.option;
pred : Base.string Base.list;
succ : Base.string Base.list;
tensor : ('a, 'b) Tensor.t Base.option;

}

Type encapsulating parameters for operations. For Convolutions and Pooling, kernel size, padding, strides For Transpose, shape

val compare : ('a, 'b) t -> ('a, 'b) t -> Base.int

val hash : ('a, 'b) t -> Base.int

val equal : ('a, 'b) t -> ('a, 'b) t -> Base.bool

val create : 
  id:Base.int ->
  name:Base.string Base.option ->
  sh:shape ->
  op:operator ->
  op_p:operator_parameters Base.option ->
  pred:Base.string Base.list ->
  succ:Base.string Base.list ->
  tensor:('a, 'b) Tensor.t Base.option ->
  ('a, 'b) t

val get_name : ('a, 'b) t -> Base.string

val get_shape : ('a, 'b) t -> shape

val get_op : ('a, 'b) t -> operator

val get_pred_list : ('a, 'b) t -> Base.string Base.list

val get_succ_list : ('a, 'b) t -> Base.string Base.list

val get_tensor : ('a, 'b) t -> ('a, 'b) Tensor.t Base.option

val is_data_node : ('a, 'b) t -> Base.bool

val is_input_node : ('a, 'b) t -> Base.bool

val is_output_node : ('a, 'b) t -> Base.bool

val num_neurons : ('a, 'b) t -> Base.int

val show : ('a, 'b) t -> ('a -> Base.string) -> Base.string

package caisar-ir