Add handling of ifs to shape propagation (#62914)
authorElias Ellison <eellison@fb.com>
Tue, 17 Aug 2021 18:21:50 +0000 (11:21 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 17 Aug 2021 18:26:42 +0000 (11:26 -0700)
commitfcc840eae048b35c811c630e961065348fcf672d
treee5a3abbabe10ea062b213e6dd5869bd1d63df82c
parent3975c08e5d73cbb43ee356ad3c91f130451e40fb
Add handling of ifs to shape propagation (#62914)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62914

Test Plan: Imported from OSS

Reviewed By: gchanan

Differential Revision: D30196945

Pulled By: eellison

fbshipit-source-id: 1c0c7f938c4547330fd1dba8ab7dd0b99a79b6a9
test/jit/test_symbolic_shape_analysis.py
torch/csrc/jit/passes/shape_analysis.cpp
torch/csrc/jit/passes/shape_analysis.h
torch/csrc/jit/passes/symbolic_shape_analysis.cpp