This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Add support for fast variable-length LSTM#14208

Merged

szha merged 36 commits intoapache:masterfrom stephenrawls:rnn-var-seqlen

May 7, 2019

+343-117

Commits on May 6, 2019

initial commit for variable length sequence support w/ cudnn
stephenrawls
committed
removing check about all vectors on same context (need to add back in)
stephenrawls
committed
fixing commented-out code to actually coment-out what I wanted
stephenrawls
committed
fixing cudnn layout type to be unpacked in var-length seq case
stephenrawls
committed
looks like param.batch_size_ etc weren't previousy getting set in cudnn operator code. still doesn't fix cudnn error though
stephenrawls
committed
must call cudnnSetRNNPaddingMode() to enable unpacked padded sequences
stephenrawls
committed
cleaning up & adding unit tests
stephenrawls
committed
cleanign up
stephenrawls
committed
cleanign up
stephenrawls
committed
removing stringstream and checking for cudnn >= 7.2
stephenrawls
committed
fixing whitespace formatting errors; adding ifdef version guard for cudnn padding
stephenrawls
committed
fixing a few syntax errors
stephenrawls
committed
changing order of arguments in hybird_forward for backward compatibility
stephenrawls
committed
more build validation fixes
stephenrawls
committed
using emplace_back to make linter happy
stephenrawls
committed
adding import of mxnet.ndarray
stephenrawls
committed
switching order of sequence_length in hybrid_forward again
stephenrawls
committed
adding __call__ override to rnn layer to handle optional sequence_length argument
stephenrawls
committed
whoops swapped order of args in one place but not the other
stephenrawls
committed
changing type() to isinstance() to make linter happy
stephenrawls
committed
changing lstm var seq length call to explciitly name sequence_length parameter
stephenrawls
committed
fixing bad scope of if-statement checking state outputs
stephenrawls
committed
resolving reviewer comments
stephenrawls
committed
making linter happy by putting var definitions in appropriate ifdef
stephenrawls
committed
fixing linter again
stephenrawls
committed
fixing whitespace issues with linter
stephenrawls
committed
fixing whitespace issues with linter
stephenrawls
committed
fixing some typos that emerged fixing linter
stephenrawls
committed
linter
stephenrawls
committed
fixing more whitespace issues
stephenrawls
committed
only access kTempSpace if on gpu
stephenrawls
committed
removing tabs that slipped in
stephenrawls
committed
fixing too-long line
stephenrawls
committed

Commits on May 7, 2019

changing ifdef guard to be more generic
stephenrawls
committed
reverting change so whitespace stays same w/ master
stephenrawls
committed
adding todo comment
stephenrawls
committed