Fix ONNX_ATEN mode (#14239)
authorzrphercule <zrphercule@gmail.com>
Wed, 21 Nov 2018 21:12:18 +0000 (13:12 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 21 Nov 2018 21:15:23 +0000 (13:15 -0800)
commit2cacb39a21216729213b877cebd97a89312fb691
treee97ca860ce668962c4d06319149683bfb99f69ee
parentfe068d903219b46d53d0665dce51be654196c0d2
Fix ONNX_ATEN mode (#14239)

Summary:
Fix ONNX_ATEN mode by adding it to the validateBlock method.
Before this pr, validateBlock will throw an exception when using this mode.

I will add related test cases for ONNX_ATEN mode in a different pr once this is merged, since we dont have any currently.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14239

Differential Revision: D13145443

Pulled By: zrphercule

fbshipit-source-id: 60e7942aa126acfe67bdb428ef231ac3066234b1
torch/csrc/jit/export.cpp