Searched refs:id_node (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 1355 Node** id_node = &identity_nodes[index]; in RemoveListArrayConverter() local
1356 if (*id_node != nullptr) { in RemoveListArrayConverter()
1362 *id_node = identity("input", {e->src(), e->src_output()}); in RemoveListArrayConverter()
1386 Node* id_node = identity_nodes[e->src_output()]; in RemoveListArrayConverter() local
1387 if (id_node == nullptr) { in RemoveListArrayConverter()
1392 CHECK(id_node); in RemoveListArrayConverter()
1393 g->AddEdge(id_node, 0, e->dst(), e->dst_input()); in RemoveListArrayConverter()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 480 const NodeDef* id_node = node_map.GetNode("id"); in TEST_F() local
481 ASSERT_NE(id_node, nullptr); in TEST_F()
482 EXPECT_EQ(1, id_node->input_size()); in TEST_F()
483 EXPECT_EQ(HoistMulName("Add_6"), id_node->input(0)); in TEST_F()
580 const NodeDef* id_node = node_map.GetNode("id"); in TEST_F() local
581 ASSERT_NE(id_node, nullptr) << "Id node not found"; in TEST_F()
582 EXPECT_EQ("id", id_node->name()); in TEST_F()
583 EXPECT_EQ(HoistMulName("add"), id_node->input(0)); in TEST_F()
660 const NodeDef* id_node = node_map.GetNode("id"); in TEST_F() local
661 ASSERT_TRUE(id_node != nullptr) << "Id node not found"; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 763 Node* id_node = graph->AddNode(ndef, &s); in BuildIdentityNode() local
765 return id_node; in BuildIdentityNode()
|