Home
last modified time | relevance | path

Searched refs:test_layer_norm_backward (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_meta.py1415 def test_layer_norm_backward(self, output_mask): member in TestMeta
H A Dtest_nestedtensor.py3431 def test_layer_norm_backward(self, device, size): member in TestNestedTensorAutograd
H A Dtest_mps.py2593 def test_layer_norm_backward(self): member in TestMPS