projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 763b995) | patch
Supply static shape info to Reshape when doing onnxGetCompatibility (#15242)
authorYinghai Lu <yinghai@fb.com>
Sat, 15 Dec 2018 00:34:11 +0000 (16:34 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 15 Dec 2018 00:37:39 +0000 (16:37 -0800)
commit717496e6c19b3c5f263f5b7836de3892d9802274
tree412d2b341f309868de2b6c4f51221bef8166fcd7tree | snapshot
parent763b9954f3fbe0b2058eee9fe8055dfdc58ee615commit | diff
Supply static shape info to Reshape when doing onnxGetCompatibility (#15242)

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

Newer version ONNX Reshape gets shape info from a tensor. Hence for static backend, we need to provide this info to it when doing `onnxGetCompatibility` too.

Reviewed By: jackm321

Differential Revision: D13471959

fbshipit-source-id: 8a58e28edd900b6ad54a1dbd63ff2579fbe0e820
caffe2/opt/onnxifi_transformer.cc diff | blob | history
Domain: Machine Learning / ML Framework;