| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer_test.cc | 78 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
81 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
137 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
140 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
186 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
189 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
243 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
246 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
247 test::ExpectTensorEqual<float>(tensors_expected[1], tensors[1]); in TEST_F()
248 test::ExpectTensorEqual<int>(tensors_expected[2], tensors[2]); in TEST_F()
[all …]
|
| D | arithmetic_optimizer_test.cc | 121 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
122 ASSERT_EQ(tensors_expected.size(), 2); in TEST_F()
150 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
166 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
167 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
182 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
200 auto tensors_expected = EvaluateNodes(item.graph, fetch); in TEST_F() local
201 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
228 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
248 auto tensors_expected = EvaluateNodes(item.graph, fetch); in TEST_F() local
[all …]
|
| D | pin_to_host_optimizer_test.cc | 60 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
67 EXPECT_EQ(tensors_expected.size(), tensors.size()); in TEST_F()
70 test::ExpectTensorEqual<int32>(tensors[i], tensors_expected[i]); in TEST_F()
72 test::ExpectTensorEqual<tstring>(tensors[i], tensors_expected[i]); in TEST_F()
100 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
107 EXPECT_EQ(tensors_expected.size(), tensors.size()); in TEST_F()
109 test::ExpectTensorEqual<float>(tensors[i], tensors_expected[i]); in TEST_F()
136 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
147 EXPECT_EQ(tensors_expected.size(), tensors.size()); in TEST_F()
149 test::ExpectTensorEqual<int32>(tensors[i], tensors_expected[i]); in TEST_F()
[all …]
|
| D | remapper_test.cc | 61 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
62 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
70 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
105 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
106 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
109 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-3); in TEST_F()
210 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, item.feed); in TEST_F() local
211 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
214 test::ExpectClose(tensors[0], tensors_expected[0], 1e-2, /*rtol=*/1e-2); in TEST_F()
322 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, item.feed); in TEST_F() local
[all …]
|
| D | common_subgraph_elimination_test.cc | 77 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
78 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
96 test::ExpectTensorNear<double>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
115 auto tensors_expected = in TEST_F() local
117 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
135 test::ExpectTensorNear<double>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
148 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
149 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
174 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
|
| D | auto_mixed_precision_test.cc | 182 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, feed); in TestSimpleUnaryInferOp() local
198 EXPECT_EQ(tensors.size(), tensors_expected.size()); in TestSimpleUnaryInferOp()
201 test::ExpectClose(tensors_expected[i], tensors[i], atol, rtol); in TestSimpleUnaryInferOp()
221 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
239 EXPECT_EQ(tensors.size(), tensors_expected.size()); in TEST_F()
242 test::ExpectTensorNear<float>(tensors_expected[i], tensors[i], 1e-6); in TEST_F()
259 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
277 EXPECT_EQ(tensors.size(), tensors_expected.size()); in TEST_F()
280 test::ExpectTensorNear<float>(tensors_expected[i], tensors[i], 1e-6); in TEST_F()
302 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
[all …]
|
| D | shape_optimizer_test.cc | 48 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
69 EXPECT_NEAR(tensors_expected[0].scalar<int>()(), in TEST_F()
71 EXPECT_NEAR(tensors_expected[1].scalar<int>()(), in TEST_F()
110 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
128 EXPECT_NEAR(tensors_expected[0].scalar<int>()(), in TEST_F()
144 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
160 EXPECT_NEAR(tensors_expected[0].scalar<int>()(), in TEST_F()
|
| D | constant_folding_test.cc | 119 auto tensors_expected = in SimpleNeutralElementTest() local
122 EXPECT_EQ(4, tensors_expected.size()); in SimpleNeutralElementTest()
125 test::ExpectTensorEqual<T>(tensors_expected[i], tensors[i]); in SimpleNeutralElementTest()
269 auto tensors_expected = in PaddingWithZeroSize() local
271 EXPECT_EQ(1, tensors_expected.size()); in PaddingWithZeroSize()
275 test::ExpectTensorEqual<int>(tensors_expected[0], tensors[0]); in PaddingWithZeroSize()
304 auto tensors_expected = EvaluateNodes(item.graph, fetch); in TEST_F() local
306 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
308 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
971 auto tensors_expected = EvaluateNodes( in TEST_F() local
[all …]
|
| D | mkl_remapper_test.cc | 174 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, item.feed); in FuseConv2DWithBiasAndAddNOrAdd() local
176 EXPECT_EQ(1, tensors_expected.size()); in FuseConv2DWithBiasAndAddNOrAdd()
180 test::ExpectClose(tensors_expected[0], tensors[0], 0, 1e-6); in FuseConv2DWithBiasAndAddNOrAdd()
301 auto tensors_expected = \
303 ASSERT_EQ(tensors_expected.size(), 1); \
306 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-6); \
441 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, item.feed); in TEST_F() local
442 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
445 test::ExpectTensorNear<float>(tensors[0], tensors_expected[0], 1e-6); in TEST_F()
521 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, item.feed); in TEST_F() local
[all …]
|
| D | memory_optimizer_test.cc | 290 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
293 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
340 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
344 test::ExpectTensorEqual<float>(tensors_expected[i], tensors[i]); in TEST_F()
389 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
392 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
477 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
480 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
508 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
511 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
[all …]
|
| D | meta_optimizer_test.cc | 396 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
401 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
402 test::ExpectTensorEqual<int>(tensors_expected[1], tensors[1]); in TEST_F()
488 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
493 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
593 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
598 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
599 test::ExpectTensorEqual<float>(tensors_expected[1], tensors[1]); in TEST_F()
982 auto tensors_expected = EvaluateNodes(item.graph, item.fetch, {}); in TEST_F() local
1016 ASSERT_EQ(tensors_expected.size(), 2); in TEST_F()
[all …]
|
| D | loop_optimizer_test.cc | 846 auto tensors_expected = EvaluateNodes(item.graph, {"m8", "m9"}); in TEST_F() local
847 ASSERT_EQ(tensors_expected.size(), 2); in TEST_F()
852 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
853 test::ExpectTensorNear<float>(tensors_expected[1], tensors[1], 1e-6); in TEST_F()
1471 auto tensors_expected = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
1472 ASSERT_EQ(tensors_expected.size(), 1); in TEST_F()
1480 test::ExpectTensorEqual<int32>(tensors_got[0], tensors_expected[0]); in TEST_F()
|
| D | generic_layout_optimizer_test.cc | 669 auto tensors_expected = EvaluateFetchNodes(item); in TEST_F() local
673 ASSERT_EQ(tensors_expected.size(), 2); in TEST_F()
674 test::ExpectTensorEqual<float>(tensors_expected[0], tensors[0]); in TEST_F()
675 test::ExpectTensorEqual<float>(tensors_expected[1], tensors[1]); in TEST_F()
|
| D | layout_optimizer_test.cc | 256 auto tensors_expected = in TEST_F() local
259 EXPECT_EQ(1, tensors_expected.size()); in TEST_F()
261 test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); in TEST_F()
|