diff options
Diffstat (limited to 'src/Analysis/Identity.hs')
| -rw-r--r-- | src/Analysis/Identity.hs | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/Analysis/Identity.hs b/src/Analysis/Identity.hs index 9dc8811..b3a6664 100644 --- a/src/Analysis/Identity.hs +++ b/src/Analysis/Identity.hs @@ -244,6 +244,12 @@ idana env expr = case expr of res <- VIArr <$> genId <*> pure sh pure (res, EMinimum1Inner res e1') + EReshape _ dim e1 e2 -> do + (v1, e1') <- idana env e1 + (_, e2') <- idana env e2 + res <- VIArr <$> genId <*> shidsToVec dim v1 + pure (res, EReshape res dim e1' e2') + EFold1InnerD1 _ cm e1 e2 e3 -> do let t1 = typeOf e2 x1 <- genIds t1 |