projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 016f212) | patch
Moderate the dim type after LengthsRangeFill (#17096)
authorYinghai Lu <yinghai@fb.com>
Thu, 14 Feb 2019 21:33:52 +0000 (13:33 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 14 Feb 2019 22:28:27 +0000 (14:28 -0800)
commitd52862ca81e419ea718447adec7e407734bf4ec8
treef97ce62f180b40f4fcc80405b3218f1461deebf8tree | snapshot
parent016f21235721d9998b1ebcd5c7948e1049233771commit | diff
Moderate the dim type after LengthsRangeFill (#17096)

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

LengthsRangeFill will take a batch size of lengths input and expand it into sequence. Later op should follow this type until it hits another batch type moderating op, e.g. SparseLengthsSum.

Reviewed By: ipiszy

Differential Revision: D14079422

fbshipit-source-id: 1a26925d502c32875ea95c160268bf6a256cc955
caffe2/opt/bound_shape_inference_test.cc diff | blob | history
caffe2/opt/bound_shape_inferencer.cc diff | blob | history
Domain: Machine Learning / ML Framework;