Searched refs:packed_enforce_sorted (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/test/nn/ |
H A D | test_packed_sequence.py | 321 packed_enforce_sorted = rnn_utils.pack_sequence([a, b, c], enforce_sorted=True) 322 self.assertEqual(packed_enforce_sorted.batch_sizes, [3, 2, 1]) 323 self.assertEqual(packed_enforce_sorted.data.data, expected) 324 self.assertTrue(packed_enforce_sorted.sorted_indices is None) 325 self.assertTrue(packed_enforce_sorted.unsorted_indices is None)
|
/aosp_15_r20/external/pytorch/test/cpp/api/ |
H A D | nn_utils.cpp | 590 rnn_utils::PackedSequence packed_enforce_sorted = in TEST_F() local 593 packed_enforce_sorted.batch_sizes(), torch::tensor({3, 2, 1}))); in TEST_F() 594 ASSERT_TRUE(torch::allclose(packed_enforce_sorted.data(), expected)); in TEST_F() 595 ASSERT_FALSE(packed_enforce_sorted.sorted_indices().defined()); in TEST_F() 596 ASSERT_FALSE(packed_enforce_sorted.unsorted_indices().defined()); in TEST_F()
|