projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 7c1e425) | patch
Ignore unknown_shaped tensor in bound shape inference (#16916)
authorYinghai Lu <yinghai@fb.com>
Tue, 12 Feb 2019 22:43:43 +0000 (14:43 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 12 Feb 2019 23:02:49 +0000 (15:02 -0800)
commit65b49b46966abc1fd9b7ad6668a994e9a669be96
tree502e2d53e032066adc2d388d56187d833270b7cetree | snapshot
parent7c1e4258a958863ee5363373289d05418be06887commit | diff
Ignore unknown_shaped tensor in bound shape inference (#16916)

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

Two fixes for maximum effort bound shape inference
1. Ignore failed and unknown shape
2. Add specialization for `SparseLengthsWeightedSumFused8BitRowwise`.

Reviewed By: ipiszy

Differential Revision: D14017810

fbshipit-source-id: 25cd68d35aa20b9ed077bdb562eb7f9deff0ab96
caffe2/opt/bound_shape_inferencer.cc diff | blob | history
caffe2/opt/bound_shape_inferencer.h diff | blob | history
Domain: Machine Learning / ML Framework;