Home
last modified time | relevance | path

Searched refs:expected_counts (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dsets_test.py178 expected_counts = [2, 0]
190 self.assertAllEqual(expected_counts, self._set_intersection_count(a, sp_b))
201 self.assertAllEqual(expected_counts,
215 expected_counts = [2, 0]
227 self.assertAllEqual(expected_counts, self._set_intersection_count(a, b))
240 expected_counts = [1]
252 self.assertAllEqual(expected_counts, self._set_intersection_count(a, b))
263 self.assertAllEqual(expected_counts, self._set_intersection_count(a, sp_b))
274 self.assertAllEqual(expected_counts,
395 expected_counts = [
[all …]
/aosp_15_r20/external/pytorch/test/cpp/profiler/
H A Drecord_function.cpp207 std::vector<int> expected_counts; in TEST() local
211 expected_counts.push_back(running_count); in TEST()
213 expected_counts.push_back(++running_count); in TEST()
228 for (const auto i : c10::irange(expected_counts.size())) { in TEST()
230 EXPECT_EQ(sample_test_counter, expected_counts[i]); in TEST()
/aosp_15_r20/external/pytorch/test/
H A Dtest_sort_and_select.py901 expected_counts = torch.tensor([1], device=device)
904 self.assertEqual(counts, expected_counts)
911 expected_counts = torch.tensor([], dtype=torch.long, device=device)
914 self.assertEqual(counts, expected_counts)
924 expected_counts, argument
944 self.assertEqual(expected_counts, ret[count_index])
953 self.assertEqual(expected_counts, y_counts)
975 expected_counts = torch.tensor([5, 3], dtype=torch.long, device=device)
980 expected_counts = torch.tensor([1, 3, 2, 1, 1], device=device)
997 expected_counts,
[all …]
H A Dtest_mps.py4100 expected_counts = torch.tensor([1], device=device)
4103 self.assertEqual(counts, expected_counts)
4110 expected_counts = torch.tensor([], dtype=torch.long, device=device)
4113 self.assertEqual(counts, expected_counts)
4115 …xpects(self, device, dtype, f, x, expected_unique, expected_inverse, expected_counts, additional_s… argument
4131 self.assertEqual(expected_counts, ret[count_index])
4138 self.assertEqual(expected_counts, y_counts)
4151 expected_counts = torch.tensor([5, 3], dtype=torch.long, device=device)
4156 expected_counts = torch.tensor([1, 3, 2, 1, 1], device=device)
4166 …ue_with_expects(device, dtype, f, x, expected_unique, expected_inverse, expected_counts, (2, 2, 2))
/aosp_15_r20/external/webrtc/test/scenario/
H A Dvideo_stream_unittest.cc64 std::vector<int> expected_counts; in TEST() local
66 expected_counts.push_back( in TEST()
69 EXPECT_GE(frame_counts[0], expected_counts[0]); in TEST()
70 EXPECT_GE(frame_counts[1], expected_counts[1]); in TEST()