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

Adapt pp_relu_linear to CAMUS network

parent b3656556
......@@ -428,7 +428,7 @@ let pp_relu_linear tname n env t g =
let idx_1 = Base.List.nth_exn idx 3 in
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 idx_0) idx_1
(Base.List.nth_exn to_activate idx_1) 0
with
(* match false with *)
| true -> "(assert (= |CELL_"^tname^idx_str^"| |CELL_"^pname^idx_str^"|))"
......
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