|
|
@@ -11,7 +11,7 @@ let rec expand dims depth = function
|
|
|
| [node] -> dim_reduce depth node
|
|
|
| hd :: tl -> let trav = dim_reduce depth in
|
|
|
let mul = Binop (Mul, trav hd, trav (List.hd dims), noloc) in
|
|
|
- Binop (Mul, mul, expand (List.tl dims) depth tl, noloc)
|
|
|
+ Binop (Add, mul, expand (List.tl dims) depth tl, noloc)
|
|
|
|
|
|
and dim_reduce depth = function
|
|
|
| Allocate (name, dims, dec, loc) ->
|