Searched refs:expected_tensors (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner_test.cc | 88 auto expected_tensors = EvaluateNodes(item.graph, fetch); in TEST_F() local
90 ASSERT_EQ(expected_tensors.size(), 1); in TEST_F()
92 test::ExpectTensorEqual<float>(actual_tensors[0], expected_tensors[0]); in TEST_F()
130 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
131 ASSERT_EQ(expected_tensors.size(), 1); in TEST_F()
132 test::ExpectTensorEqual<float>(actual_tensors[0], expected_tensors[0]); in TEST_F()
178 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
179 ASSERT_EQ(expected_tensors.size(), 2); in TEST_F()
181 test::ExpectTensorEqual<float>(actual_tensors[i], expected_tensors[i]); in TEST_F()
225 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
[all …]
|
| D | constant_folding_test.cc | 1120 auto expected_tensors = in TEST_F() local
1125 EXPECT_EQ(fetch.size(), expected_tensors.size()); in TEST_F()
1132 test::ExpectTensorNear<float>(expected_tensors[i], tensors[i], 1e-5); in TEST_F()
1189 auto expected_tensors = in TEST_F() local
1193 EXPECT_EQ(1, expected_tensors.size()); in TEST_F()
1198 test::ExpectTensorNear<float>(expected_tensors[0], tensors[0], 1e-5); in TEST_F()
4384 std::vector<Tensor> expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
4403 test::ExpectTensorEqual<float>(expected_tensors[i], actual_tensors[i]); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 245 std::vector<Tensor> expected_tensors, in ExpectEqual() argument
247 if (produced_tensors.size() != expected_tensors.size()) { in ExpectEqual()
250 " v.s. ", expected_tensors.size(), ")")); in ExpectEqual()
254 if (produced_tensors[0].dtype() != expected_tensors[0].dtype()) { in ExpectEqual()
257 produced_tensors[0].dtype(), " v.s. ", expected_tensors[0].dtype(), in ExpectEqual()
268 std::sort(expected_tensors.begin(), expected_tensors.end(), \ in ExpectEqual()
298 expected_tensors[i])); in ExpectEqual()
|
| D | dataset_test_base.h | 568 std::vector<Tensor> expected_tensors,
|
| /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]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | parse_example_dataset_test.py | 57 def _compare_output_to_expected(self, dict_tensors, expected_tensors): argument
58 self.assertEqual(set(dict_tensors.keys()), set(expected_tensors.keys()))
61 expected_v = expected_tensors[k]
|