Searched refs:shared_init_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 75 const string& shared_init_name = "group_deps") { in TestSinglePartition() argument
140 CreateNode(shared_init_name, "NoOp", {}, &graph_def); in TestSinglePartition()
142 CreateNode(shared_init_name, "NoOp", {assign_node}, &graph_def, true); in TestSinglePartition()
174 if (shared_init_name != "group_deps") { in TestSinglePartition()
175 context.params["group_init_node"] = {shared_init_name}; in TestSinglePartition()
258 EXPECT_EQ(1, node_lookup.count(shared_init_name)); in TestSinglePartition()
259 EXPECT_EQ("NoOp", node_lookup.at(shared_init_name)->op()); in TestSinglePartition()
278 EXPECT_NE(std::find(node_lookup.at(shared_init_name)->input().begin(), in TestSinglePartition()
279 node_lookup.at(shared_init_name)->input().end(), in TestSinglePartition()
281 node_lookup.at(shared_init_name)->input().end()); in TestSinglePartition()
[all …]
|