Searched defs:LayerNorm (Results 1 – 11 of 11) sorted by relevance
65 class LayerNorm(NodeVisitor): class
42 class LayerNorm(torch.nn.Module): class in TestLayerNorm
29 class LayerNorm(nn.Module): class
373 layernorm: Optional[nn.LayerNorm],
94 class LayerNorm(Module): class
14 class LayerNorm(torch.nn.LayerNorm): class
156 class LayerNorm(jit.ScriptModule): class
572 class LayerNorm(torch.nn.Module): class
1413 TEST_F(FunctionalTest, LayerNorm) { in TEST_F() argument
1125 TEST_F(ModulesTest, LayerNorm) { in TEST_F() argument
665 TEST(StaticRuntime, LayerNorm) { in TEST() argument