Commit a881ed7e authored by Aymeric Varasse's avatar Aymeric Varasse
Browse files

Adapt pp_relu_linear to CAMUS network

parent 79393d1c
......@@ -420,16 +420,16 @@ let pp_relu_linear tname n env t g =
in
let is_positive x = if (x >= 0.) then true else false in
let is_positive_tuple (x0, x1, x2) = [is_positive x0; is_positive x1; is_positive x2] in
let to_activate = Base.List.concat_map ~f:is_positive_tuple n_op in
let to_activate = Base.List.map ~f:is_positive_tuple n_op in
let relu_cell_to_smt idx =
List.iter print_int idx;
let idx_str = stringify_int idx in
let idx_0 = Base.List.nth_exn idx 0 in
let idx_1 = Base.List.nth_exn idx 1 in
match Base.List.nth_exn to_activate (idx_1 * (Base.Int.of_float(Base.Float.sqrt (Base.Float.of_int(List.length n_op))))
(* + idx_0 * (Base.Int.of_float(Base.Float.sqrt (Base.Float.of_int(List.length n_op)))) *)
+ idx_0 * 3
) with
match Base.List.nth_exn
(* (idx_1 * (Base.Int.of_float(Base.Float.sqrt (Base.Float.of_int(List.length n_op)))) + idx_0 * (Base.Int.of_float(Base.Float.sqrt (Base.Float.of_int(List.length n_op)))) + idx_0 * 3) *)
(Base.List.nth_exn to_activate 0) idx_0
with
(* match false with *)
| true -> "(assert (= |CELL_"^tname^idx_str^"| |CELL_"^pname^idx_str^"|))"
| false -> begin match t with
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment