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()
|