Lines Matching full:allclose
74 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
85 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
107 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
118 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
133 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
158 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
169 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
194 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
204 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
229 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
266 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
283 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
301 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
331 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
353 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
375 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
402 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
430 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper()
478 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper_gelu()
495 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper_gelu()
513 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper_gelu()
543 ASSERT_TRUE(torch::allclose( in transformer_decoder_layer_test_helper_gelu()
604 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_test_helper()
615 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_test_helper()
640 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_test_helper()
666 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_test_helper()
691 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_test_helper()
719 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_test_helper()
745 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_test_helper()
864 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
881 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
899 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
929 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
951 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
973 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1000 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1028 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1047 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1082 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1103 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1139 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1160 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1177 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1195 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1225 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1259 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1296 ASSERT_TRUE(torch::allclose( in transformer_decoder_test_helper()
1441 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_test_helper()
1457 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_test_helper()
1488 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_test_helper()