Module `Nier_cfg.Node`Source

Sourcetype shape = Base.int Base.array

Sourcetype 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.

Sourceval str_op : operator -> Base.string

Sourceval show_shape : shape -> Base.string

Sourcetype ksize =

| Ksize of shape

Sourcetype stride =

| Stride of shape

Sourcetype pads =

| Pads of shape

Sourcetype dilations =

| Dilations of shape

Sourcetype 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)

Sourceval str_op_params : operator_parameters -> Base.string

Sourcetype ('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

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

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

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

Source

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

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

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

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

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

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

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

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

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

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

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

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

Install

dune-project
Dependency

Authors

Maintainers

Sources

doc/caisar.ir/Ir/Nier_cfg/Node/index.html

Module `Nier_cfg.Node`Source

package caisar

Install

dune-project Dependency

Authors

Maintainers

Sources

doc/caisar.ir/Ir/Nier_cfg/Node/index.html

Module Nier_cfg.NodeSource

dune-project
Dependency

Module `Nier_cfg.Node`Source