Searched refs:test_memory_layout (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_torch.py | 8968 def test_memory_layout(x, y, scale, zero_point, out): function
8991 test_memory_layout(xraw, yraw, None, None, xraw + yraw)
8992 test_memory_layout(qxraw, qyraw, 0.1, 5, torch.ops.quantized.add(qxraw, qyraw, 0.1, 5))
8997 test_memory_layout(x, y, None, None, x + y)
9000 test_memory_layout(qx, qy, 0.1, 5, torch.ops.quantized.add(qx, qy, 0.1, 5))
9005 test_memory_layout(x, y, None, None, x + y)
9008 test_memory_layout(qx, qy, 0.1, 5, torch.ops.quantized.add(qx, qy, 0.1, 5))
9019 test_memory_layout(x, y, None, None, torch.add(x, y, out=out))
9025 test_memory_layout(x, y, None, None, x + y)
9028 test_memory_layout(qx, qy, 0.1, 5, torch.ops.quantized.add(qx, qy, 0.1, 5))
|