Home
last modified time | relevance | path

Searched refs:layernorm_pytorch_lstm_creator (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Drunner.py11 layernorm_pytorch_lstm_creator,
72 "cudnn_layernorm", layernorm_pytorch_lstm_creator, DummyContext
H A Dfactory.py325 def layernorm_pytorch_lstm_creator(**kwargs): function