Home
last modified time | relevance | path

Searched defs:decompose_layernorm (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Dcustom_lstms.py79 decompose_layernorm=False, argument
183 def __init__(self, input_size, hidden_size, decompose_layernorm=False): argument
H A Dfactory.py81 def lnlstm_creator(script=True, decompose_layernorm=False, **kwargs): argument