107     ASSERT_TRUE(torch::allclose(p1.grad(), p2.grad()));  in TEST_F()
257         ASSERT_TRUE(torch::allclose(  in TEST_F()
345   ASSERT_TRUE(torch::allclose(p1.grad(), p2.grad()));  in TEST_F()
359   ASSERT_TRUE(vector.allclose(expected));  in TEST_F()
368     ASSERT_TRUE(zero_parameters[i].allclose(parameters[i]));  in TEST_F()
434   ASSERT_TRUE(torch::allclose(a.data(), b.data()));  in assert_is_equal_packed_sequence()
435   ASSERT_TRUE(torch::allclose(a.batch_sizes(), b.batch_sizes()));  in assert_is_equal_packed_sequence()
438       torch::allclose(a.sorted_indices(), b.sorted_indices()));  in assert_is_equal_packed_sequence()
441       torch::allclose(a.unsorted_indices(), b.unsorted_indices()));  in assert_is_equal_packed_sequence()
500       ASSERT_TRUE(torch::allclose(lengths, lengths_out));  in TEST_F()
516       ASSERT_TRUE(torch::allclose(unpacked, ref_output));  in TEST_F()
559     ASSERT_TRUE(torch::allclose(padded, std::get<0>(unpacked)));  in TEST_F()
572   ASSERT_TRUE(torch::allclose(packed.batch_sizes(), torch::tensor({3, 2, 1})));  in TEST_F()
573   ASSERT_TRUE(torch::allclose(packed.data(), expected));  in TEST_F()
575       torch::allclose(packed.sorted_indices(), torch::tensor({0, 1, 2})));  in TEST_F()
577       torch::allclose(packed.unsorted_indices(), torch::tensor({0, 1, 2})));  in TEST_F()
582       torch::allclose(packed_unsorted.batch_sizes(), torch::tensor({3, 2, 1})));  in TEST_F()
583   ASSERT_TRUE(torch::allclose(packed_unsorted.data(), expected));  in TEST_F()
584   ASSERT_TRUE(torch::allclose(  in TEST_F()
586   ASSERT_TRUE(torch::allclose(  in TEST_F()
592   ASSERT_TRUE(torch::allclose(  in TEST_F()
594   ASSERT_TRUE(torch::allclose(packed_enforce_sorted.data(), expected));  in TEST_F()
744       ASSERT_TRUE(torch::allclose(packed.data(), expected_data));  in TEST_F()
745       ASSERT_TRUE(torch::allclose(packed.batch_sizes(), batch_sizes));  in TEST_F()
749           torch::allclose(packed.unsorted_indices(), unsorted_indices));  in TEST_F()
754       ASSERT_TRUE(torch::allclose(unpacked, src));  in TEST_F()
755       ASSERT_TRUE(torch::allclose(unpacked_len, lengths));  in TEST_F()
773         ASSERT_TRUE(torch::allclose(  in TEST_F()
821   ASSERT_TRUE(padded.allclose(expected));  in TEST_F()
825   ASSERT_TRUE(padded.allclose(expected.transpose(0, 1)));  in TEST_F()
830   ASSERT_TRUE(padded.allclose(expected));  in TEST_F()
834   ASSERT_TRUE(padded.allclose(expected.transpose(0, 1)));  in TEST_F()
839   ASSERT_TRUE(padded.allclose(expected));  in TEST_F()
844   ASSERT_TRUE(padded.allclose(expected));  in TEST_F()
871     ASSERT_TRUE(padded.allclose(expected));  in TEST_F()
875     ASSERT_TRUE(padded.allclose(expected.transpose(0, 1)));  in TEST_F()
887     ASSERT_TRUE(padded.allclose(expected));  in TEST_F()
891     ASSERT_TRUE(padded.allclose(expected.transpose(0, 1)));  in TEST_F()