Searched refs:test_layer_norm_backward (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_meta.py | 1415 def test_layer_norm_backward(self, output_mask): member in TestMeta |
H A D | test_nestedtensor.py | 3431 def test_layer_norm_backward(self, device, size): member in TestNestedTensorAutograd |
H A D | test_mps.py | 2593 def test_layer_norm_backward(self): member in TestMPS |