| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/fft/ |
| H A D | test_helper.py | 18 from numpy.testing import assert_array_almost_equal
22 from torch._numpy.testing import assert_array_almost_equal
29 assert_array_almost_equal(fft.fftshift(x), y)
30 assert_array_almost_equal(fft.ifftshift(y), x)
33 assert_array_almost_equal(fft.fftshift(x), y)
34 assert_array_almost_equal(fft.ifftshift(y), x)
39 assert_array_almost_equal(fft.ifftshift(fft.fftshift(x)), x)
44 assert_array_almost_equal(fft.fftshift(freqs, axes=(0, 1)), shifted)
45 assert_array_almost_equal(
48 assert_array_almost_equal(fft.ifftshift(shifted, axes=(0, 1)), freqs)
[all …]
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/ |
| H A D | test_index_tricks.py | 29 assert_array_almost_equal,
46 assert_array_almost_equal,
245 assert_array_almost_equal(y, mgrid[2:10:50j], 13)
254 assert_array_almost_equal(c[0][-1, :], np.ones(10, "d"), 11)
255 assert_array_almost_equal(c[1][:, -1], 2 * np.ones(10, "d"), 11)
256 assert_array_almost_equal(d[0, 1, :] - d[0, 0, :], 0.1 * np.ones(20, "d"), 11)
257 assert_array_almost_equal(d[1, :, 1] - d[1, :, 0], 0.2 * np.ones(20, "d"), 11)
293 assert_array_almost_equal(grid64, grid32)
299 assert_array_almost_equal(grid64, grid32)
307 assert_array_almost_equal(grid64, grid128)
[all …]
|
| H A D | test_histograms.py | 29 assert_array_almost_equal,
40 assert_array_almost_equal,
180 assert_array_almost_equal(a * 5, wa)
181 assert_array_almost_equal(na, nwa)
187 assert_array_almost_equal(wa, w)
193 assert_array_almost_equal(wa, np.array([4, 5, 0, 1]) / 10.0 / 3.0 * 4)
215 assert_array_almost_equal(wa, np.array([1, 1]) + 1j * np.array([2, 3]))
219 assert_array_almost_equal(wa, np.array([1, 1]) + 1j * np.array([2, 3]))
229 assert_array_almost_equal(wa, [Decimal(1), Decimal(5)])
233 assert_array_almost_equal(wa, [Decimal(1), Decimal(5)])
[all …]
|
| H A D | test_twodim_base.py | 45 assert_array_almost_equal,
70 assert_array_almost_equal,
261 assert_array_almost_equal(H, answer / 8.0, 3)
270 assert_array_almost_equal(H, answer, 3)
|
| H A D | test_function_base.py | 73 assert_array_almost_equal,
105 assert_array_almost_equal,
1189 assert_array_almost_equal(y, yo, 11)
1190 assert_array_almost_equal(z, zo, 11)
1353 assert_array_almost_equal(r1, r2)
1933 assert_array_almost_equal(w, np.flipud(w), 7)
2073 assert_array_almost_equal(c, np.array([[1.0, -1.0], [-1.0, 1.0]]))
|
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_kernels.py | 113 np.testing.assert_array_almost_equal(
120 np.testing.assert_array_almost_equal(
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_module.py | 217 np.testing.assert_array_almost_equal(
226 np.testing.assert_array_almost_equal(
332 np.testing.assert_array_almost_equal(
364 np.testing.assert_array_almost_equal(
386 np.testing.assert_array_almost_equal(
395 np.testing.assert_array_almost_equal(
1478 np.testing.assert_array_almost_equal(
1500 np.testing.assert_array_almost_equal(
1509 np.testing.assert_array_almost_equal(
|
| H A D | test_quantized_functional.py | 66 np.testing.assert_array_almost_equal(
239 np.testing.assert_array_almost_equal(
|
| H A D | test_quantized_op.py | 3875 … np.testing.assert_array_almost_equal(Y_q_ref, Y_q.int_repr().numpy(), decimal=decimal_val)
3889 np.testing.assert_array_almost_equal(
4042 … np.testing.assert_array_almost_equal(Y_fp32_ref.numpy(), Y_q_dq.numpy(), decimal=decimal_val)
4137 np.testing.assert_array_almost_equal(Y_q_ref, Y_q.int_repr().numpy(), decimal=decimal_val)
4175 np.testing.assert_array_almost_equal(np.float32(W_q.q_per_channel_scales().numpy()),
4457 np.testing.assert_array_almost_equal(
4912 np.testing.assert_array_almost_equal(
5112 np.testing.assert_array_almost_equal(
6848 np.testing.assert_array_almost_equal(
7865 np.testing.assert_array_almost_equal(a_pool_q.int_repr().numpy(),
[all …]
|
| H A D | test_quantized_tensor.py | 1369 np.testing.assert_array_almost_equal(X_scale, qparams[0], decimal=3)
|
| /aosp_15_r20/external/pytorch/torch/_numpy/testing/ |
| H A D | __init__.py | 8 assert_array_almost_equal,
|
| H A D | utils.py | 435 return assert_array_almost_equal(actual, desired, decimal, err_msg)
839 def assert_array_almost_equal(x, y, decimal=6, err_msg="", verbose=True): function
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
| H A D | test_numeric.py | 45 assert_array_almost_equal,
57 assert_array_almost_equal,
2413 assert_array_almost_equal(z, self.z1)
2415 assert_array_almost_equal(z, self.z1_4)
2417 assert_array_almost_equal(z, self.z2)
2419 assert_array_almost_equal(z, self.z1r)
2421 assert_array_almost_equal(z, self.z2r)
2423 assert_array_almost_equal(z, self.zs)
2438 assert_array_almost_equal(z, r_z)
2467 assert_array_almost_equal(np.convolve(d, k)[2:-2], np.full(98, 3))
|
| H A D | test_multiarray.py | 47 assert_array_almost_equal,
62 assert_array_almost_equal,
2995 assert_array_almost_equal(np.array([0.5]) * np.array([1.0, 2.0]), [0.5, 1.0])
2998 assert_array_almost_equal(d * (d * np.array([1.0, 2.0])), [0.25, 0.5])
3900 assert_array_almost_equal(x[~nan_mask], y[~nan_mask], decimal=5)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_tensorboard.py | 212 np.testing.assert_array_almost_equal(np.sum(x), np.sum(y))
222 np.testing.assert_array_almost_equal(np.sum(x), np.sum(y))
|