Lines Matching full:allclose
26 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
29 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
48 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
51 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
68 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
71 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
106 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
109 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
117 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
126 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
144 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
153 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
162 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
171 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
181 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
187 ASSERT_TRUE(torch::allclose( in TEST_F()
198 ASSERT_TRUE(torch::allclose(y1, torch::ones({2, 2, 2, 2}))); in TEST_F()
204 ASSERT_TRUE(torch::allclose( in TEST_F()
219 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
225 ASSERT_TRUE(torch::allclose( in TEST_F()
248 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
266 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
284 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
294 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
305 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
320 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
337 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
351 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
363 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
376 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
389 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
402 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
417 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
431 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
450 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
460 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
468 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
474 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
483 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 3}))); in TEST_F()
492 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 3, 3}))); in TEST_F()
501 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 3, 3, 3}))); in TEST_F()
510 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 3}))); in TEST_F()
519 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 3, 3}))); in TEST_F()
528 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 3, 3, 3}))); in TEST_F()
584 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
605 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
616 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
621 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
632 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
643 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
654 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
672 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
673 ASSERT_TRUE(output_aligned.allclose(expected_aligned)); in TEST_F()
697 ASSERT_TRUE(output.allclose(expected, 1e-2)); in TEST_F()
698 ASSERT_TRUE(output_aligned.allclose(expected_aligned)); in TEST_F()
766 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
777 ASSERT_TRUE(output.allclose(expected, 1e-4)); in TEST_F()
789 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
802 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
817 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
856 ASSERT_TRUE(distanceOutput.allclose(basicOutput, 1e-6, 1e-6)); in TEST_F()
881 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
882 ASSERT_TRUE(F::nll_loss(input, target).allclose(expected, 1e-04)); in TEST_F()
894 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
895 ASSERT_TRUE(F::cross_entropy(input, target).allclose(expected, 1e-04)); in TEST_F()
905 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
915 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
925 ASSERT_TRUE(torch::allclose( in TEST_F()
935 ASSERT_TRUE(torch::allclose( in TEST_F()
949 ASSERT_TRUE(torch::allclose( in TEST_F()
961 torch::allclose(y, torch::tensor({{{0, 2, 0, 4, 5}}}, torch::kFloat))); in TEST_F()
978 ASSERT_TRUE(torch::allclose( in TEST_F()
1006 ASSERT_TRUE(torch::allclose( in TEST_F()
1030 ASSERT_TRUE(torch::allclose( in TEST_F()
1045 ASSERT_TRUE(torch::allclose( in TEST_F()
1073 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1074 ASSERT_TRUE(torch::allclose(y_bf16.to(torch::kFloat), y, 1e-2, 1e-2)); in TEST_F()
1076 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1077 ASSERT_TRUE(torch::allclose(x_bf16.to(torch::kFloat), y, 1e-2, 1e-2)); in TEST_F()
1098 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1099 ASSERT_TRUE(output_bf16.to(torch::kFloat).allclose(output, 1e-2, 1e-2)); in TEST_F()
1101 ASSERT_TRUE(input.allclose(expected)); in TEST_F()
1102 ASSERT_TRUE(input_bf16.to(torch::kFloat).allclose(output, 1e-2, 1e-2)); in TEST_F()
1110 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1124 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1125 ASSERT_TRUE(F::glu(input).allclose(expected)); in TEST_F()
1132 ASSERT_TRUE(torch::allclose(y, y_exp, 1.4e-06, 1e-05)); in TEST_F()
1140 ASSERT_TRUE(torch::allclose(y, y_exp, 1.4e-06, 1e-05)); in TEST_F()
1157 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1169 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
1185 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
1199 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
1220 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1222 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1246 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1248 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1267 ASSERT_TRUE(torch::allclose(y, y_exp, 1e-4, 1e-7)); in TEST_F()
1282 ASSERT_TRUE(torch::allclose( in TEST_F()
1298 ASSERT_TRUE(torch::allclose( in TEST_F()
1313 ASSERT_TRUE(torch::allclose( in TEST_F()
1333 ASSERT_TRUE(torch::allclose( in TEST_F()
1376 ASSERT_TRUE(torch::allclose(output[i], expected)); in TEST_F()
1388 ASSERT_TRUE(torch::allclose(output[i], expected)); in TEST_F()
1400 ASSERT_TRUE(torch::allclose(output[i], expected)); in TEST_F()
1410 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1418 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1426 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1439 ASSERT_TRUE(torch::allclose(y, y_exp, 1e-4, 1e-7)); in TEST_F()
1455 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1468 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1478 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1495 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1507 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1520 ASSERT_TRUE(torch::allclose(y_with_bias, y_with_bias_exp, 1e-4, 1e-7)); in TEST_F()
1526 ASSERT_TRUE(torch::allclose(y_no_bias, y_no_bias_exp, 1e-4, 1e-7)); in TEST_F()
1553 ASSERT_TRUE(torch::allclose(norm, expected)); in TEST_F()
1565 ASSERT_TRUE(torch::allclose(output, expected)); in TEST_F()
1588 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1590 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1598 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1600 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1615 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1628 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1630 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1638 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1640 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1655 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1677 ASSERT_TRUE(torch::allclose(z, torch::ones_like(z))); in TEST_F()
1679 ASSERT_TRUE(torch::allclose(x, y)); in TEST_F()
1703 ASSERT_TRUE(torch::allclose(z, torch::ones_like(z))); in TEST_F()
1723 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1724 ASSERT_TRUE(torch::allclose(y_bf16.to(torch::kFloat), y, 1e-2, 1e-2)); in TEST_F()
1726 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1727 ASSERT_TRUE(torch::allclose(x_bf16.to(torch::kFloat), y, 1e-2, 1e-2)); in TEST_F()
1747 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1748 ASSERT_TRUE(torch::allclose(y_bf16.to(torch::kFloat), y, 1e-2, 1e-2)); in TEST_F()
1765 ASSERT_TRUE(y.allclose(y_exp)); in TEST_F()
1782 ASSERT_TRUE(y.allclose(y_exp)); in TEST_F()
1799 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1816 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1829 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1848 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1866 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1891 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1899 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1907 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1923 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1925 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
1955 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1964 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1991 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2006 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2033 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2048 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2081 ASSERT_TRUE(output.allclose(expected, 2e-04)); in TEST_F()
2098 ASSERT_TRUE(output.allclose(expected, 2e-04)); in TEST_F()
2132 ASSERT_TRUE(output.allclose(expected, 2e-04)); in TEST_F()
2152 ASSERT_TRUE(output.allclose(expected, 2e-04)); in TEST_F()
2196 ASSERT_TRUE(output.allclose(expected, 2e-04)); in TEST_F()
2226 ASSERT_TRUE(output.allclose(expected, 2e-04)); in TEST_F()
2239 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2257 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2277 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2341 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2355 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2369 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2381 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2399 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2445 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2475 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2515 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2547 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2612 ASSERT_TRUE(torch::allclose( in TEST_F()
2628 ASSERT_TRUE(torch::allclose( in TEST_F()
2642 ASSERT_TRUE(torch::allclose( in TEST_F()
2644 ASSERT_TRUE(torch::allclose( in TEST_F()
2650 ASSERT_TRUE(torch::allclose( in TEST_F()
2656 ASSERT_TRUE(torch::allclose( in TEST_F()
2669 ASSERT_TRUE(torch::allclose( in TEST_F()
2678 ASSERT_TRUE(torch::allclose( in TEST_F()
2692 ASSERT_TRUE(torch::allclose( in TEST_F()
2715 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
2718 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
2748 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
2751 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
2781 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
2784 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
2799 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
2802 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
2817 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.1)); in TEST_F()
2818 ASSERT_TRUE(torch::allclose(input_std, output.std(), 0.1)); in TEST_F()
2820 ASSERT_TRUE(torch::allclose(input_, output)); in TEST_F()
2825 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.1)); in TEST_F()
2826 ASSERT_TRUE(torch::allclose(input_std, output.std(), 0.1)); in TEST_F()
2841 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.1)); in TEST_F()
2842 ASSERT_TRUE(torch::allclose(input_std, output.std(), 0.1)); in TEST_F()
2844 ASSERT_TRUE(torch::allclose(input_, output)); in TEST_F()
2849 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.1)); in TEST_F()
2850 ASSERT_TRUE(torch::allclose(input_std, output.std(), 0.1)); in TEST_F()
2860 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.01, 0.05)); in TEST_F()
2864 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.01, 0.05)); in TEST_F()
2876 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.01, 0.05)); in TEST_F()
2879 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.01, 0.05)); in TEST_F()
2890 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.01, 0.05)); in TEST_F()
2893 ASSERT_TRUE(torch::allclose(input_mean, output.mean(), 0.01, 0.05)); in TEST_F()
2923 ASSERT_TRUE(torch::allclose( in test_isfinite()
2924 // torch::allclose does not support comparing torch::kBool in test_isfinite()
2994 ASSERT_TRUE(torch::allclose( in test_isinf()
2995 // torch::allclose does not support comparing torch::kBool in test_isinf()
3053 ASSERT_TRUE(torch::allclose(x, x)); in test_allclose()
3054 ASSERT_TRUE(torch::allclose(x, y)); in test_allclose()
3055 ASSERT_TRUE(torch::allclose(y, x)); in test_allclose()
3056 ASSERT_FALSE(torch::allclose(1.1 * x + 0.1, 1.0 * x)); in test_allclose()
3057 ASSERT_TRUE(torch::allclose(0.99 * x + 0.1, 1.0 * x, 1.1, 0.1)); in test_allclose()
3065 ASSERT_TRUE(torch::allclose(x, x)); in test_allclose()
3066 ASSERT_TRUE(torch::allclose(x, y)); in test_allclose()
3067 ASSERT_TRUE(torch::allclose(y, x)); in test_allclose()
3076 ASSERT_TRUE(torch::allclose(x, x, 1.0, 0.0, /*equal_nan=*/true)); in test_allclose()
3077 ASSERT_TRUE(torch::allclose(x, y, 1.0, 0.0, /*equal_nan=*/true)); in test_allclose()
3078 ASSERT_TRUE(torch::allclose(y, x, 1.0, 0.0, /*equal_nan=*/true)); in test_allclose()
3087 ASSERT_TRUE(torch::allclose(x, x, 1.0, 0.0, /*equal_nan=*/true)); in test_allclose()
3088 ASSERT_TRUE(torch::allclose(x, y, 1.0, 0.0, /*equal_nan=*/true)); in test_allclose()
3089 ASSERT_TRUE(torch::allclose(y, x, 1.0, 0.0, /*equal_nan=*/true)); in test_allclose()
3093 TEST_F(FunctionalTest, AllClose) { in TEST_F() argument
3139 ASSERT_TRUE(torch::allclose( in TEST_F()
3144 ASSERT_TRUE(torch::allclose( in TEST_F()
3157 ASSERT_TRUE(torch::allclose( in TEST_F()
3161 ASSERT_TRUE(torch::allclose( in TEST_F()
3173 ASSERT_TRUE(torch::allclose( in TEST_F()
3192 ASSERT_TRUE(torch::allclose(output.grad(), expected_grad)); in TEST_F()
3210 ASSERT_TRUE(torch::allclose(out1, out2)); in TEST_F()
3224 ASSERT_TRUE(torch::allclose(out1, out2)); in TEST_F()
3231 ASSERT_TRUE(torch::allclose( in TEST_F()
3260 ASSERT_TRUE(torch::allclose(out1, out2)); in TEST_F()
3261 ASSERT_TRUE(torch::allclose(out1, out3)); in TEST_F()
3277 ASSERT_TRUE(torch::allclose(grad, expected_grad)); in TEST_F()