Add simplify pass in the Relay interpreter (#2417)

This commit is contained in:
Haichen Shen 2019-01-11 13:57:00 -08:00 коммит произвёл Tianqi Chen
Родитель 547a0913f6
Коммит 312802f341
1 изменённых файлов: 3 добавлений и 1 удалений

Просмотреть файл

@ -218,7 +218,9 @@ class Interpreter(Executor):
"""
# TODO: We need to move this optimization code into the optimizer/pass manager
ck_expr = ir_pass.infer_type(expr, mod=self.mod)
fused_expr = ir_pass.fuse_ops(ck_expr)
simp_expr = ir_pass.simplify_inference(ck_expr)
ck_simp = ir_pass.infer_type(simp_expr, mod=self.mod)
fused_expr = ir_pass.fuse_ops(ck_simp)
ck_fused = ir_pass.infer_type(fused_expr, mod=self.mod)
return ck_fused