Searched refs:test_multiple_inputs (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ | ||
H A D | special_math_ops_test.py | 772 def test_multiple_inputs(self): member in EinsumTest |
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_legacy_vmap.py | 84 def test_multiple_inputs(self): member in TestVmapAPILegacy |
/aosp_15_r20/external/pytorch/test/functorch/ | ||
H A D | test_eager_transforms.py | 2531 def test_multiple_inputs(self, device): member in TestJvp |
H A D | test_vmap.py | 168 def test_multiple_inputs(self): member in TestVmapAPI |