projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 0c237f1) | patch
Add shape inference function for Split (#18838)
authorYinghai Lu <yinghai@fb.com>
Thu, 4 Apr 2019 07:19:21 +0000 (00:19 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 4 Apr 2019 07:22:22 +0000 (00:22 -0700)
commite5e2110a8ead028c863a7f449273bf6ee90bc423
tree6bb667e33b825feae1ea56eb8cefadd8c4f529eatree | snapshot
parent0c237f1383abdddcfe7dcaa00af7fc075ddb8d67commit | diff
Add shape inference function for Split (#18838)

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

It turns out that we don't have shape inference function of `Split` op at all. This diff adds that.

Reviewed By: bertmaher

Differential Revision: D14766871

fbshipit-source-id: 535cb4f24bdada603c76579e00e7a39aee93e19f
caffe2/operators/concat_split_op.cc diff | blob | history
caffe2/opt/bound_shape_inference_test.cc diff | blob | history
Domain: Machine Learning / ML Framework;