Home
last modified time | relevance | path

Searched refs:packed_enforce_sorted (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_packed_sequence.py321 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 Dnn_utils.cpp590 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()