| /external/tensorflow/tensorflow/core/common_runtime/ |
| parallel_concat_optimizer.cc | 107 Node* identity_node; variable
108 TF_RETURN_IF_ERROR(identity_def.Finalize(g, &identity_node));
113 g->AddControlEdge(identity_node, e->dst());
115 g->AddEdge(identity_node, 0, e->dst(), e->dst_input());
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_nodes_test.cc | 142 NodeDef* identity_node = graph_def.add_node(); local
143 identity_node->set_name("identity_node");
144 identity_node->set_op("Identity");
145 identity_node->add_input("add_node");
150 context.output_names = {"identity_node"};
160 EXPECT_EQ(1, node_lookup.count("identity_node"));
161 EXPECT_EQ("add_node", node_lookup.at("identity_node")->input(0));
199 context.output_names = {"identity_node"};
|
| sparsify_gather_test.cc | 135 NodeDef* identity_node = local
137 MakeGather("gather", gather_v2, identity_node, input_node, &graph_def);
154 {identity_node, concat_input_node, concat_axis_node}, &graph_def);
160 {identity_node, concat_input_node,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| vectorization_utils_test.cc | 1373 const NodeDef& identity_node = vectorized->node_def( local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| mutable_graph_view.cc | 999 NodeDef* identity_node = GetNode(identity_name); local
[all...] |
