Searched refs:expected_tensors (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner_test.cc | 95 auto expected_tensors = EvaluateNodes(item.graph, fetch); in TEST_F() local
97 EXPECT_EQ(1, expected_tensors.size()); in TEST_F()
99 test::ExpectTensorEqual<float>(expected_tensors[0], actual_tensors[0]); in TEST_F()
140 auto expected_tensors = EvaluateNodes(item.graph, fetch); in TEST_F() local
142 EXPECT_EQ(1, expected_tensors.size()); in TEST_F()
144 test::ExpectTensorEqual<float>(expected_tensors[0], actual_tensors[0]); in TEST_F()
198 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
200 EXPECT_EQ(2, expected_tensors.size()); in TEST_F()
202 for (int i = 0; i < expected_tensors.size(); i++) { in TEST_F()
203 test::ExpectTensorEqual<float>(expected_tensors[i], actual_tensors[i]); in TEST_F()
[all …]
|
| D | constant_folding_test.cc | 806 auto expected_tensors = in TEST_F() local
811 EXPECT_EQ(fetch.size(), expected_tensors.size()); in TEST_F()
818 test::ExpectTensorNear<float>(expected_tensors[i], tensors[i], 1e-5); in TEST_F()
875 auto expected_tensors = in TEST_F() local
879 EXPECT_EQ(1, expected_tensors.size()); in TEST_F()
884 test::ExpectTensorNear<float>(expected_tensors[0], tensors[0], 1e-5); in TEST_F()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | parse_example_dataset_test.py | 55 def _compare_output_to_expected(self, dict_tensors, expected_tensors): argument
56 self.assertEqual(set(dict_tensors.keys()), set(expected_tensors.keys()))
59 expected_v = expected_tensors[k]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_conditional_accumulator_test.py | 169 expected_tensors = []
175 expected_tensors.append(expected_tensor)
181 self._assertEqual_indexedslices(expected_tensors[i], result)
|
| D | parse_single_example_op_test.py | 69 def _compare_output_to_expected(tester, dict_tensors, expected_tensors, argument
71 tester.assertEqual(set(dict_tensors.keys()), set(expected_tensors.keys()))
75 expected_v = expected_tensors[k]
|
| D | parsing_ops_test.py | 69 def _compare_output_to_expected(tester, dict_tensors, expected_tensors, argument
71 tester.assertEqual(set(dict_tensors.keys()), set(expected_tensors.keys()))
75 expected_v = expected_tensors[k]
|