Module Nier_cfg.NodeSource

Sourcetype operator =
  1. | Add
  2. | Sub
  3. | Mul
  4. | Div
  5. | Matmul
  6. | Gemm
  7. | LogSoftmax
  8. | ReLu
  9. | Transpose
  10. | Squeeze
  11. | MaxPool
  12. | Conv
  13. | Reshape
  14. | Flatten
  15. | Identity
  16. | Constant
  17. | NO_OP
  18. | RW_Linearized_ReLu

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

Sourceval str_op : operator -> Base.string
Sourceval show_shape : shape -> Base.string
Sourcetype ksize =
  1. | Ksize of shape
Sourcetype stride =
  1. | Stride of shape
Sourcetype pads =
  1. | Pads of shape
Sourcetype dilations =
  1. | Dilations of shape
Sourcetype operator_parameters =
  1. | Pool_params of ksize * stride Base.option * pads Base.option * dilations Base.option
  2. | Conv_params of ksize * stride Base.option * pads Base.option * dilations Base.option
  3. | Transpose_params of shape
  4. | 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 = {
  1. id : Base.int;
  2. name : Base.string Base.option;
  3. shape : shape;
  4. operator : operator;
  5. operator_parameters : operator_parameters Base.option;
  6. pred : Base.string Base.list;
  7. succ : Base.string Base.list;
  8. 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
Sourceval 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