Searched refs:expected_nodes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph_test.cc | 80 std::vector<string> expected_nodes = str_util::Split(nodes, ','); in ExpectNodes() local
81 std::sort(expected_nodes.begin(), expected_nodes.end()); in ExpectNodes()
82 for (const string& s : expected_nodes) { in ExpectNodes()
90 EXPECT_TRUE(actual_nodes.size() == expected_nodes.size()) in ExpectNodes()
92 << "\nExpected: " << absl::StrJoin(expected_nodes, ","); in ExpectNodes()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer_test.py | 86 expected_nodes = [
90 self.assertEqual(len(optimized_graph_nodes), len(expected_nodes))
91 self.assertAllInSet(optimized_graph_nodes, expected_nodes)
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model_test.cc | 200 size_t expected_nodes = use_resource ? 4 : 3; in TestFreezeGraphWithDependentVariables() local
201 EXPECT_EQ(frozen_graph_def.node_size(), expected_nodes); in TestFreezeGraphWithDependentVariables()
259 size_t expected_nodes = use_resource ? 4 : 3; in TestFreezeGraphWithAndWithoutDependentVariables() local
260 EXPECT_EQ(frozen_graph_def.node_size(), expected_nodes); in TestFreezeGraphWithAndWithoutDependentVariables()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 1322 std::vector<string> expected_nodes = {"dflt", "p1", "p2", "i3"}; in TEST_F() local
1323 EXPECT_EQ(output.node_size(), expected_nodes.size()); in TEST_F()
1327 EXPECT_EQ(expected_nodes[i], output.node(i).name()); in TEST_F()
1365 std::vector<string> expected_nodes = {"dflt", "p1", "p2", "c", in TEST_F() local
1367 EXPECT_EQ(output.node_size(), expected_nodes.size()); in TEST_F()
1371 EXPECT_EQ(expected_nodes[i], output.node(i).name()); in TEST_F()
1423 std::vector<string> expected_nodes = {"dflt", "p1", "p2", "i2"}; in TEST_F() local
1424 EXPECT_EQ(output.node_size(), expected_nodes.size()); in TEST_F()
1427 EXPECT_EQ(expected_nodes[i], output.node(i).name()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 745 std::vector<string> expected_nodes = {"cluster1", "cluster2", "mul", "x"}; in TEST() local
746 EXPECT_EQ(expected_nodes, GraphNodes(*graph)); in TEST()
|