Lines Matching full:allclose
53 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
92 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
114 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
168 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
202 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
237 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
273 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
293 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
309 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
320 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
324 torch::allclose(indices, torch::tensor({{{0, 2}}}, torch::kLong))); in TEST_F()
336 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
349 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
360 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
362 ASSERT_TRUE(torch::allclose( in TEST_F()
376 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
387 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
390 ASSERT_TRUE(torch::allclose( in TEST_F()
407 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
420 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
433 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
446 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
459 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
470 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
472 ASSERT_TRUE(torch::allclose( in TEST_F()
485 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
496 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2, 2}))); in TEST_F()
499 ASSERT_TRUE(torch::allclose( in TEST_F()
521 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
539 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
557 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
644 ASSERT_TRUE(torch::allclose(y, torch::tensor({{{2, 4, 5}}}, torch::kFloat))); in TEST_F()
656 ASSERT_TRUE(torch::allclose(y, torch::tensor({{{2, 4, 5}}}, torch::kFloat))); in TEST_F()
659 torch::allclose(indices, torch::tensor({{{1, 3, 4}}}, torch::kLong))); in TEST_F()
672 ASSERT_TRUE(torch::allclose( in TEST_F()
693 ASSERT_TRUE(torch::allclose( in TEST_F()
716 ASSERT_TRUE(torch::allclose( in TEST_F()
727 ASSERT_TRUE(torch::allclose( in TEST_F()
749 ASSERT_TRUE(torch::allclose( in TEST_F()
760 ASSERT_TRUE(torch::allclose( in TEST_F()
782 ASSERT_TRUE(torch::allclose( in TEST_F()
805 ASSERT_TRUE(torch::allclose( in TEST_F()
817 ASSERT_TRUE(torch::allclose( in TEST_F()
841 torch::allclose(y, torch::tensor({{{1.5, 3.0, 4.5}}}, torch::kFloat))); in TEST_F()
856 ASSERT_TRUE(torch::allclose( in TEST_F()
878 ASSERT_TRUE(torch::allclose( in TEST_F()
900 ASSERT_TRUE(torch::allclose( in TEST_F()
920 ASSERT_TRUE(torch::allclose( in TEST_F()
932 torch::allclose(y, torch::tensor({{{0, 2, 0, 4, 5}}}, torch::kFloat))); in TEST_F()
941 ASSERT_TRUE(torch::allclose( in TEST_F()
948 ASSERT_TRUE(torch::allclose( in TEST_F()
951 ASSERT_TRUE(torch::allclose( in TEST_F()
969 ASSERT_TRUE(torch::allclose( in TEST_F()
993 ASSERT_TRUE(torch::allclose( in TEST_F()
999 ASSERT_TRUE(torch::allclose( in TEST_F()
1018 ASSERT_TRUE(torch::allclose( in TEST_F()
1038 ASSERT_TRUE(torch::allclose( in TEST_F()
1075 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1092 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1121 ASSERT_TRUE(torch::allclose(y, y_exp, 1e-4, 1e-7)); in TEST_F()
1140 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1158 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
1192 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1224 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1302 ASSERT_TRUE(torch::allclose( in TEST_F()
1311 ASSERT_TRUE(torch::allclose( in TEST_F()
1366 ASSERT_TRUE(y.allclose(x)); in TEST_F()
1395 ASSERT_TRUE(y.allclose(x)); in TEST_F()
1425 ASSERT_TRUE(y.allclose(x)); in TEST_F()
1541 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1604 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1677 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1740 ASSERT_TRUE(output.allclose(expected, 1e-3)); in TEST_F()
1804 ASSERT_TRUE(output.allclose(expected, 1e-3)); in TEST_F()
1878 ASSERT_TRUE(output.allclose(expected, 1e-3)); in TEST_F()
1978 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
1994 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2010 ASSERT_TRUE(output.allclose(expected, 1e-4)); in TEST_F()
2025 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2039 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2053 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2067 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2081 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2095 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2110 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2125 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2142 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2181 ASSERT_TRUE(distanceOutput.allclose(basicOutput, 1e-6, 1e-6)); in TEST_F()
2183 distanceOperatorOutput.allclose(distanceOutput, 1e-6, 1e-6)); in TEST_F()
2185 distanceOperatorOutput.allclose(basicOperatorOutput, 1e-6, 1e-6)); in TEST_F()
2246 ASSERT_TRUE(moduleOutput.allclose(functionOutput, 1e-6, 1e-6)); in TEST_F()
2248 moduleOperatorOutput.allclose(functionOutput, 1e-6, 1e-6)); in TEST_F()
2268 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2272 .allclose(expected, 1e-04)); in TEST_F()
2285 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2291 .allclose(expected, 1e-04)); in TEST_F()
2304 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2318 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2333 ASSERT_TRUE(output.allclose(expected, 1e-04)); in TEST_F()
2347 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2363 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2378 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2397 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2412 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2437 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2439 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
2465 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2467 ASSERT_TRUE(input.allclose(expected)); in TEST_F()
2488 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2515 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2517 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
2562 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2564 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
2588 ASSERT_TRUE(torch::allclose(y, y_exp, 1e-4, 1e-7)); in TEST_F()
2599 ASSERT_TRUE(torch::allclose(output[i], expected)); in TEST_F()
2611 ASSERT_TRUE(torch::allclose(output[i], expected)); in TEST_F()
2623 ASSERT_TRUE(torch::allclose(output[i], expected)); in TEST_F()
2635 torch::allclose(torch::exp(logprob_out).data().sum(1), torch::ones(4))); in TEST_F()
2646 ASSERT_TRUE(torch::allclose(predict_out, logprob_out.argmax(1))); in TEST_F()
2672 ASSERT_TRUE(torch::allclose(loss, expected)); in TEST_F()
2673 ASSERT_TRUE(torch::allclose( in TEST_F()
2686 torch::allclose(asfm(x, y).output.squeeze(0), asfm(x2, y2).output)); in TEST_F()
2713 ASSERT_TRUE(torch::allclose(output[i][j][k][l], expected)); in TEST_F()
2728 torch::allclose(model->weight, torch::full(num_parameters, init))); in TEST_F()
2740 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2760 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2762 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
2787 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2789 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
2820 ASSERT_TRUE(torch::allclose(z, torch::ones_like(z))); in TEST_F()
2822 ASSERT_TRUE(torch::allclose(x, y)); in TEST_F()
2852 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2854 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
2875 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
2878 ASSERT_TRUE(model_default_options->forward(input).allclose(expected)); in TEST_F()
2886 ASSERT_TRUE(torch::allclose(y, y_exp, 1.4e-06, 1e-05)); in TEST_F()
2895 ASSERT_TRUE(torch::allclose(y, y_exp, 1.4e-06, 1e-05)); in TEST_F()
2905 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2914 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2932 ASSERT_TRUE(y.allclose(y_exp)); in TEST_F()
2950 ASSERT_TRUE(y.allclose(y_exp)); in TEST_F()
2967 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2987 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
2997 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
3006 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
3015 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
3033 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
3035 ASSERT_TRUE(torch::allclose(x, y_exp)); in TEST_F()
3054 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3073 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3088 ASSERT_TRUE(torch::allclose(output.narrow(2, 0, 15), expected)); in TEST_F()
3104 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3124 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3146 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3164 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3185 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3202 ASSERT_TRUE(torch::allclose( in TEST_F()
3213 torch::allclose(component_wise_loss, loss->forward(input, target))); in TEST_F()
3217 ASSERT_TRUE(torch::allclose( in TEST_F()
3222 ASSERT_TRUE(torch::allclose( in TEST_F()
3233 ASSERT_TRUE(torch::allclose( in TEST_F()
3244 ASSERT_TRUE(torch::allclose( in TEST_F()
3255 ASSERT_TRUE(torch::allclose( in TEST_F()
3282 ASSERT_TRUE(torch::allclose( in TEST_F()
3287 ASSERT_TRUE(torch::allclose( in TEST_F()
3295 ASSERT_TRUE(torch::allclose( in TEST_F()
3299 ASSERT_TRUE(torch::allclose( in TEST_F()
3306 ASSERT_TRUE(torch::allclose( in TEST_F()
3318 ASSERT_TRUE(torch::allclose(output.grad(), expected_grad)); in TEST_F()
3332 ASSERT_TRUE(torch::allclose(out1, out2)); in TEST_F()
3342 ASSERT_TRUE(torch::allclose(out1, out2)); in TEST_F()
3349 ASSERT_TRUE(torch::allclose( in TEST_F()
3369 ASSERT_TRUE(torch::allclose(out1, out2)); in TEST_F()
3370 ASSERT_TRUE(torch::allclose(out1, out3)); in TEST_F()
3383 ASSERT_TRUE(torch::allclose(grad, expected_grad)); in TEST_F()
3766 torch::allclose(result, reference, 1e-5, 1e-5, /*equal_nan=*/true)); in _multihead_attn_test_helper()
3771 ASSERT_TRUE(torch::allclose( in _multihead_attn_test_helper()
3891 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3900 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3918 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3931 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3958 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3974 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3985 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
3994 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4012 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4025 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4052 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4085 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4097 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4106 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4124 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4137 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4164 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4197 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4210 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4220 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4237 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4250 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4277 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4310 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4332 ASSERT_TRUE(input.grad().allclose(grad_expected)); in TEST_F()
4333 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4344 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4355 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4366 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4377 ASSERT_TRUE(output.allclose(expected)); in TEST_F()
4389 ASSERT_TRUE(torch::allclose(output, expected, 1e-05, 2e-04)); in TEST_F()
4394 ASSERT_TRUE(torch::allclose(output, expected, 1e-05, 2e-04)); in TEST_F()
4400 ASSERT_TRUE(torch::allclose(output, expected, 1e-05, 2e-04)); in TEST_F()
4446 ASSERT_TRUE(torch::allclose(output_hx, expected_hx, 1e-05, 2e-04)); in TEST_F()
4447 ASSERT_TRUE(torch::allclose(output_cx, expected_cx, 1e-05, 2e-04)); in TEST_F()
4456 ASSERT_TRUE(torch::allclose(output_hx, expected_hx, 1e-05, 2e-04)); in TEST_F()
4457 ASSERT_TRUE(torch::allclose(output_cx, expected_cx, 1e-05, 2e-04)); in TEST_F()
4467 ASSERT_TRUE(torch::allclose(output_hx, expected_hx, 1e-05, 2e-04)); in TEST_F()
4468 ASSERT_TRUE(torch::allclose(output_cx, expected_cx, 1e-05, 2e-04)); in TEST_F()
4534 ASSERT_TRUE(torch::allclose(output, expected, 1e-05, 2e-04)); in TEST_F()
4539 ASSERT_TRUE(torch::allclose(output, expected, 1e-05, 2e-04)); in TEST_F()
4545 ASSERT_TRUE(torch::allclose(output, expected, 1e-05, 2e-04)); in TEST_F()