Searched refs:pytorch_lstm_creator (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | runner.py | 18 pytorch_lstm_creator,
67 "cudnn": RNNRunner("cudnn", pytorch_lstm_creator, DummyContext),
69 "cudnn_dropout", partial(pytorch_lstm_creator, dropout=0.4), DummyContext
79 "aten": RNNRunner("aten", pytorch_lstm_creator, DisableCuDNN),
|
| H A D | test.py | 6 from .factory import pytorch_lstm_creator, varlen_pytorch_lstm_creator
182 test_rnns(creator, pytorch_lstm_creator, **test_args)
|
| H A D | factory.py | 58 def pytorch_lstm_creator(**kwargs): function
|