Searched refs:assert_path_equal (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
| H A D | test_einsum.py | 1212 def assert_path_equal(self, comp, benchmark): member in TestEinsumPath
1227 self.assert_path_equal(path, ["einsum_path", (0, 1, 2)])
1230 self.assert_path_equal(path, ["einsum_path", (0, 1, 2)])
1234 self.assert_path_equal(path, ["einsum_path", (0, 1, 2, 3)])
1237 self.assert_path_equal(path, ["einsum_path", (0, 1, 2, 3)])
1245 self.assert_path_equal(
1250 self.assert_path_equal(
1257 self.assert_path_equal(
1262 self.assert_path_equal(
1272 self.assert_path_equal(path, ["einsum_path", (0, 2), (0, 1)])
[all …]
|