xref: /aosp_15_r20/external/pytorch/benchmarks/fastrnns/__init__.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1from .cells import *  # noqa: F403
2from .factory import *  # noqa: F403
3
4
5# (output, next_state) = cell(input, state)
6seqLength = 100
7numLayers = 2
8inputSize = 512
9hiddenSize = 512
10miniBatch = 64
11