Searched defs:jacapi (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 1790 def test_simple(self, device, jacapi): argument
1797 def test_simple_not_flat(self, device, jacapi): argument
1805 def test_take(self, device, jacapi): argument
1816 def test_diff_numel(self, device, jacapi): argument
1833 def test_vmap_on_jac_simple(self, device, jacapi): argument
1840 def test_nested_jac_simple(self, device, jacapi): argument
1850 def test_multiple_args(self, device, jacapi): argument
1858 def test_multiple_outputs_multiple_argnums(self, device, jacapi): argument
1880 def test_multiple_outputs_single_argnums(self, device, jacapi): argument
1901 def test_multiple_outputs_pytree(self, device, jacapi): argument
[all …]
|