| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 112 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
113 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
131 test::ExpectTensorNear<double>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
150 auto tensors_expected = in TEST_F() local
152 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
171 test::ExpectTensorNear<double>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
184 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
185 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
210 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
223 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
[all …]
|
| D | pin_to_host_optimizer_test.cc | 80 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
87 EXPECT_EQ(tensors_expected.size(), tensors.size()); in TEST_F()
90 test::ExpectTensorEqual<int32>(tensors[i], tensors_expected[i]); in TEST_F()
92 test::ExpectTensorEqual<string>(tensors[i], tensors_expected[i]); in TEST_F()
119 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
130 EXPECT_EQ(tensors_expected.size(), tensors.size()); in TEST_F()
132 test::ExpectTensorEqual<int32>(tensors[i], tensors_expected[i]); in TEST_F()
159 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
166 EXPECT_EQ(tensors_expected.size(), tensors.size()); in TEST_F()
168 test::ExpectTensorEqual<int32>(tensors[i], tensors_expected[i]); in TEST_F()
[all …]
|
| D | function_optimizer_test.cc | 112 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
115 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
188 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
191 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
288 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
291 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
372 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
375 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
422 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
425 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
[all …]
|
| D | remapper_test.cc | 56 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
57 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
65 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
96 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
97 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
100 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
160 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, item.feed); in TEST_F() local
162 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
164 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
225 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, item.feed); in TEST_F() local
[all …]
|
| D | constant_folding_test.cc | 113 auto tensors_expected = in SimpleNeutralElementTest() local
116 EXPECT_EQ(4, tensors_expected.size()); in SimpleNeutralElementTest()
119 test::ExpectTensorEqual<T>(tensors_expected[i], tensors[i]); in SimpleNeutralElementTest()
244 auto tensors_expected = EvaluateNodes(item.graph, fetch); in TEST_F() local
246 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
248 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
657 auto tensors_expected = EvaluateNodes( in TEST_F() local
660 EXPECT_EQ(item.fetch.size(), tensors_expected.size()); in TEST_F()
666 test::ExpectTensorNear<float>(tensors_expected[i], tensors[i], 1e-6); in TEST_F()
978 auto tensors_expected = EvaluateNodes(item.graph, fetch); in TEST_F() local
[all …]
|
| D | shape_optimizer_test.cc | 44 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
65 EXPECT_NEAR(tensors_expected[0].scalar<int>()(), in TEST_F()
67 EXPECT_NEAR(tensors_expected[1].scalar<int>()(), in TEST_F()
83 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
99 EXPECT_NEAR(tensors_expected[0].scalar<int>()(), in TEST_F()
|
| D | memory_optimizer_test.cc | 289 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
292 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
339 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
343 test::ExpectTensorEqual<float>(tensors_expected[i], tensors[i]); in TEST_F()
388 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
391 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
476 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
479 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
507 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
510 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
[all …]
|
| D | meta_optimizer_test.cc | 398 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
403 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
404 test::ExpectTensorEqual<int>(tensors_expected[1], tensors[1]); in TEST_F()
490 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
495 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
595 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
600 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
601 test::ExpectTensorEqual<float>(tensors_expected[1], tensors[1]); in TEST_F()
|
| D | loop_optimizer_test.cc | 748 auto tensors_expected = EvaluateNodes(item.graph, {"m7", "m8", "m9"}); in TEST_F() local
749 ASSERT_EQ(tensors_expected.size(), 3); in TEST_F()
754 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
755 test::ExpectTensorNear<float>(tensors_expected[1], tensors[1], 1e-6); in TEST_F()
1374 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
1375 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
1383 test::ExpectTensorEqual<int32>(tensors_got[0], tensors_expected[0]); in TEST_F()
|
| D | layout_optimizer_test.cc | 250 auto tensors_expected = in TEST_F() local
253 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
255 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
|