Searched refs:expected_element (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | parallel_interleave_test.py | 189 for expected_element in self._interleave(
191 self.write_coordination_events[expected_element].set()
192 self.assertEqual(expected_element * expected_element,
217 for expected_element in self._interleave(
219 self.write_coordination_events[expected_element].set()
221 self.assertEqual(expected_element * expected_element, output)
242 for i, expected_element in enumerate(
245 self.write_coordination_events[expected_element].set()
247 self.read_coordination_events[expected_element].acquire()
250 self.read_coordination_events[expected_element].acquire()
[all …]
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | task_runner_test.cc | 88 for (auto& expected_element : elements) { in TEST() local
94 test::ExpectEqual(element[0], expected_element[0]); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | while_test.cc | 972 auto expected_element = LiteralUtil::CreateR1<float>({1, 1}); in XLA_TEST_F() local
974 LiteralUtil::MakeTuple({&expected_element, &expected_element}); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_comparison.cc | 782 const auto expected_element = LiteralSlice(expected, {i}); in NearHelper() local
787 NearHelper(expected_element, actual_element, element_index, error, in NearHelper()
|